题意:
一共有
n
n
n个电台,对于每个电台
i
i
i有三个参数:
x
i
,
r
i
,
f
i
x_i,r_i,f_i
xi,ri,fi ,分别指它的一维坐标、作用半径和频率。如果两个电台的频率差值在
k
k
k内,并且它们的作用范围都能覆盖到彼此,那么就称这两个电台相互干扰。
k
k
k是一开始给定的。问这
n
n
n个站台中相互干扰的站台有多少对。
n
<
=
1
e
5
,
k
<
=
10
,
x
i
,
r
i
<
=
1
e
9
,
f
i
<
=
1
e
4
n<=1e5,k<=10,x_i,r_i<=1e9,f_i<=1e4
n<=1e5,k<=10,xi,ri<=1e9,fi<=1e4。
题解:
这种题看上去就应该需要按照某一个顺序做,需要排序。那么我们考虑一下要按照什么顺序排序。我考虑了四种可能的顺序,分别是按照
x
i
x_i
xi、按照
f
i
f_i
fi、按照
x
i
x_i
xi与
r
i
r_i
ri的和或者差、按照
r
i
r_i
ri。然后你逐个考虑按照每个排序能不能做。我一开始在考虑前三种顺序的时候都口胡了假做法,并且写出来之后才发现假了,具体过程就不说了。我感觉按照前三种策略排序的话是没有什么简便的做法的,如果有什么好的想法欢迎与我交流。
那么我们最后就是以半径为关键字排序来做了。排完序我们要考虑按照半径从小往大做还是从大往小做。我们考虑按照半径排好序之后的作用,就会发现从小到大没什么用,但是从大到小就不一样了,因为可以保证之前考虑过的只要你能覆盖它,它就一定能覆盖你。那么我们现在统计答案的思路就是在每一个点加进来的时候考虑之前有多少与它产生了满足要求的数对,这个东西我们不能一一枚举去判断,那么我们就要考虑需要知道哪些量才能快速计算。
我们发现 k k k很小,于是可能正解会是一个复杂度含 k k k的东西,也就是我们可以对 k k k暴力。我们考虑与 k k k相关的是频率,也就是对于当前枚举到的点 i i i,我们可以暴力枚举与 f i f_i fi相差 k k k以内的每一个权值,看之前考虑过的所有点中权值是枚举的权值的有多少个。但是并不是出现过的都能被当前的 i i i覆盖,所以要考虑哪些才能被覆盖,不难发现能被覆盖的点是满足下标在区间 [ x i − r i , x i + r i ] [x_i-r_i,x_i+r_i] [xi−ri,xi+ri]之间的点能被记入答案。我们发现没法直接记录这个答案,于是考虑用数据结构维护。如果这个下标范围很小的话,我们只需要对于每种权值维护一个线段树,线段树的下标表示是点的下标,就变成了一个单点修改区间求和的问题了。但是权值很大,而我们发现线段树上的绝大多数节点都不会被用到,那么我们的办法就是用动态开点线段树了。
这样这个题就做完了,时间复杂度 O ( n l o g f i ∗ k ) O(nlogf_i*k) O(nlogfi∗k),空间复杂度是一个 l o g log log的。
代码:
#include <bits/stdc++.h>
using namespace std;
int n,k,cnt,root[10010];
long long ans;
const int qwq=1e9;
struct node
{
int x,r,f;
}a[100010];
struct tree
{
int l,r,sz;
}tr[8000010];
inline int read()
{
int x=0;
char s=getchar();
while(s>'9'||s<'0')
s=getchar();
while(s>='0'&&s<='9')
{
x=x*10+s-'0';
s=getchar();
}
return x;
}
inline int cmp(node x,node y)
{
return x.r>y.r;
}
inline void update(int &rt,int l,int r,int x)
{
if(!rt)
rt=++cnt;
tr[rt].sz++;
if(l==r)
return;
int mid=(l+r)>>1;
if(x<=mid)
update(tr[rt].l,l,mid,x);
else
update(tr[rt].r,mid+1,r,x);
}
inline int query(int rt,int l,int r,int le,int ri)
{
if(!rt)
return 0;
if(le<=l&&r<=ri)
return tr[rt].sz;
int mid=(l+r)>>1,res=0;
if(le<=mid)
res+=query(tr[rt].l,l,mid,le,ri);
if(mid+1<=ri)
res+=query(tr[rt].r,mid+1,r,le,ri);
return res;
}
int main()
{
n=read();
k=read();
for(int i=1;i<=n;++i)
{
a[i].x=read();
a[i].r=read();
a[i].f=read();
}
sort(a+1,a+n+1,cmp);
for(int i=1;i<=n;++i)
{
for(int j=max(1,a[i].f-k);j<=min(10000,a[i].f+k);++j)
ans+=query(root[j],1,qwq,max(1,a[i].x-a[i].r),min(qwq,a[i].x+a[i].r));
update(root[a[i].f],1,qwq,a[i].x);
}
printf("%I64d\n",ans);
return 0;
}