地址:http://codeforces.com/contest/1046/problem/A
转载:https://blog.csdn.net/Lee_w_j__/article/details/82823266
来考虑如何知道两个机器人是否能互相看见。
我们可以先按从大到小对机器人进行排序,然后用一个线段树记录第 i 个节点是否有机器人,这样我们就可以通过线段树的区间查询知道,对于第 i 个机器人,它可以看到多少个机器人。
由于我们是动态加入机器人的,同时之前加入的机器人的 r 是要大于等于当前机器人的 r 的,所以当前机器人所能看到的机器人肯定也能看到当前的机器人。
接下来我们再来考虑智商的约束条件,常规做法是线段树再套一个线段树,仍旧是通过区间查询查出合适的机器人,但由于本题的内存只有256MB,树套树的空间接受不了,所以无法用树套树解决。
解决本题有一个关键,就是k<=20,这样每次我们只需要暴力遍历40次即可,由于空间的限制原因,所以我们可以通过线段树动态开点来维护智商为 qi 的机器人所在的位置,由于总的只有1e5个机器人,所以所耗费的空间会比树套树小很多。这样就可以解决这个题了。
相当于每个iq建立一个线段树,用动态开点来压缩空间
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int inf = 0x3f3f3f3f;
const int N = 1e9;
const int MAXN = 1e5 + 7;
int root[MAXN * 40],ls[MAXN * 40],lt[MAXN * 40];
LL sum[MAXN * 40];
int n,k,cnt;
vector<int>ve;
typedef struct Node{
int x,r,q;
friend bool operator < (const Node &a,const Node &b)
{
return a.r > b.r;
}
}Node;
Node node[MAXN];
int getid(int x)
{
return lower_bound(ve.begin(),ve.end(),x) - ve.begin() + 1;
}
void update(int p,int l,int r,int &rt)
{
if(!rt) rt = ++cnt;
if(l == r){
sum[rt] += 1;
return ;
}
int mid = (l + r) / 2;
if(p <= mid) update(p,l,mid,ls[rt]);
else update(p,mid + 1,r,lt[rt]);
sum[rt] += 1;
}
LL query(int L,int R,int l,int r,int rt)
{
if(!rt) return 0;
if(L <= l && R >= r) return sum[rt];
LL ans = 0;
int mid = (l + r) / 2;
if(L <= mid) ans += query(L,R,l,mid,ls[rt]);
if(R > mid) ans += query(L,R,mid + 1,r,lt[rt]);
return ans;
}
int main()
{
while(~scanf("%d %d",&n,&k))
{
ve.clear();
memset(root,0,sizeof(root));
memset(ls,0,sizeof(ls));
memset(lt,0,sizeof(lt));
memset(sum,0,sizeof(sum));
cnt = 0;
for(int i = 0;i < n;++i)
{
scanf("%d %d %d",&node[i].x,&node[i].r,&node[i].q);
ve.push_back(node[i].q);
ve.push_back(node[i].q - k);
ve.push_back(node[i].q + k);
}
sort(node,node + n);
sort(ve.begin(),ve.end());
ve.erase(unique(ve.begin(),ve.end()),ve.end());
LL ans = 0;
for(int i = 0;i < n;++i)
{
int st = getid(node[i].q - k),sk = getid(node[i].q + k);
for(int j = st;j <= sk;++j){
if(!root[j]) continue;
int l = max(0,node[i].x - node[i].r);
int r = min(N,node[i].x + node[i].r);
ans += query(l,r,0,N,root[j]);
}
int id = getid(node[i].q);
update(node[i].x,0,N,root[id]);
}
printf("%lld\n",ans);
}
return 0;
}