CodeForces - 1046A （动态开线段树）

CodeForces - 1046A

题意：

给一组机器人，每个机器人有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;
}