CF762E Radio stations

一、题目

点此看题

二、解法

我们发现 min ( r i , r j ) \min(r_i,r_j) 很难处理,我们可以把 r r 从大到小排序,那么就可以直接用 r i r_i 来替换,问题变成了找到 i i 前面的一个点 j j ,满足 x i r i x j x i + r i x_i-r_i\leq x_j\leq x_i+r_i f i k f j f i + k f_i-k\leq f_j\leq f_i+k

这个问题的本质就是查询矩阵内点的个数,可以容斥,把每一个点的询问拆成 4 4 的询问,这样就变成了最经典的三维偏序问题,直接上模板吧!

#include <cstdio>
#include <algorithm>
using namespace std;
const int M = 100005;
int read()
{
	int x=0,flag=1;char c;
	while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
	while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
	return x*flag;
}
int n,m,k,bit[M];long long ans;
struct node
{
	int x,r,f;
	bool operator < (const node &b) const
	{
		return r>b.r;
	}
}a[M];
struct data
{
	int tp,x,y,w;
	bool operator < (const data &b) const
	{
		return x<b.x;
	}
}p[5*M];
int lowbit(int x)
{
	return x&(-x);
}
void upd(int x,int f)
{
	for(int i=x;i<=10010;i+=lowbit(i))
		bit[i]+=f;
}
int ask(int x)
{
	int r=0;
	for(int i=x;i>=1;i-=lowbit(i))
		r+=bit[i];
	return r;
}
void cdq(int l,int r)
{
	if(l==r) return ;
	int mid=(l+r)>>1,j=l;
	cdq(l,mid);cdq(mid+1,r);
	sort(p+l,p+mid+1);
	sort(p+mid+1,p+r+1);
	for(int i=mid+1;i<=r;i++)
	{
		while(j<=mid && p[i].x>=p[j].x)
		{
			if(p[j].tp==0) upd(p[j].y,1);
			j++;
		}
		if(p[i].tp==1)
			ans+=p[i].w*ask(p[i].y);
	}
	for(j--;j>=l;j--)
		if(p[j].tp==0)
			upd(p[j].y,-1);
}
int main()
{
	n=read();k=read();
	for(int i=1;i<=n;i++)
	{
		int x=read(),r=read(),f=read();
		a[i]=node{x,r,f};
	}
	sort(a+1,a+1+n);
	for(int i=1;i<=n;i++)
	{
		int x=a[i].x,r=a[i].r,f=a[i].f;
		int x1=x-r,x2=x+r,y1=max(f-k,1),y2=f+k;
		p[++m]=data{1,x1-1,y1-1,1};
		p[++m]=data{1,x1-1,y2,-1};
		p[++m]=data{1,x2,y1-1,-1};
		p[++m]=data{1,x2,y2,1};
		p[++m]=data{0,x,f,0};
	}
	cdq(1,m);
	printf("%lld\n",ans);
}
原创文章 430 获赞 14 访问量 1万+

猜你喜欢

转载自blog.csdn.net/C202044zxy/article/details/105862906