版权声明: https://blog.csdn.net/sgh666666/article/details/83623806
题意:
给一组机器人,每个机器人有x,r,q三个属性。x:坐标,r:视野范围,q:IQ值。
每个机器人视野为:[x-r,x+r],如果一对机器人相互可以看到,且它们的智商差距不大于K,那么他们可以聊天。问多少对机器人可以聊天。
题解:
先按视野降序排序,后面能看到前面的话,则前面的一定也可以看见后面。
然后对于当前机器人来说,看在他前面有几个智商在[q-k,q+k],[x-r,x+r]。然后做个矩阵覆盖就可以了。又因为智商这个一维树比较小,所以可以对每个智商动态开点线段树,然后一个个扫过去统计答案即可。
const int maxn = 4e6+11;
map<ll,ll>mp;
const ll w = 1e9+11;
ll n,k,cnt,ans;
ll x[100011],r[100011],q[100011],id[100011];
struct node{ll v,l,r;}t[maxn];
bool cmp(ll aa,ll bb){return r[aa] > r[bb];}
ll query(ll rt,ll l,ll r,ll ql,ll qr)
{
if(!rt)return 0;
if(l == ql && qr == r)return t[rt].v;
ll mid = (l + r)>>1;
if(qr <= mid)return query(t[rt].l,l,mid,ql,qr);
else if (ql > mid)return query(t[rt].r,mid+1,r,ql,qr);
return query(t[rt].l,l,mid,ql,mid) + query(t[rt].r,mid+1,r,mid+1,qr);
}
void update(ll &rt,ll l,ll r,ll p)
{
if(!rt)rt = ++cnt;
t[rt].v++;
if(l == r)return;
ll mid = (l+r)>>1;
if(p<=mid)update(t[rt].l,l,mid,p);
else update(t[rt].r,mid+1,r,p);
}
int main()
{
int n,k;
scanf("%d%d",&n,&k);
for(ll i = 1;i<=n;i++)
{
scanf("%lld%lld%lld",x+i,r+i,q+i);
id[i] = i;
}
sort(id+1,id+1+n,cmp);//按r排序
for(ll i = 1;i<=n;i++)
{
ll now = id[i];
for(ll j = q[now]-k;j<=q[now]+k;j++)
{
if(!mp[j])continue;
ans += query(mp[j],0,w,max(0ll,x[now]-r[now]),min(w,x[now]+r[now]));
}
update(mp[q[now]],0,w,x[now]);
}
printf("%lld\n",ans);
return 0;
}