codeforces 1046_problem_A AI robots (cdq分治)
阿新 • • 發佈:2018-11-05
題意:給n個機器人,機器人都有自己的位置x,可見範圍r,智商q,問兩個機器人互相可見,並且智商相差不多於k的對數有多少。
思路:運用分治思想,第一維對於r非遞減排序,分治,考慮對於左半部分的每一個機器人,右半部分有多少個和它配對的,因為r及從小到大排序,所以只要左能看見右,右就一定能看到左,
第二維對q排序,然後用樹狀陣列維護x,對於q要保證是和列舉的左邊的那個機器人智商保持在k的範圍之內。
#include<bits/stdc++.h> using namespace std; const int N=1e5+10; typedef long long ll; int k; int val[N], cnt; ll ans=0; struct Robot{ int x, r, q; int L, R; }p[N], fz[N]; bool cmp1(Robot a, Robot b){ return a.r<b.r; } ll s[N]; void upd(int x, int v){ for(;x<N; x+=x&-x){ s[x]+=v; } } ll query(int x){ ll ret=0; for(; x>0; x-=x&-x){ ret+=s[x]; } return ret; } bool cmp2(Robot a, Robot b){ return a.q<b.q; } void cdq(int l, int r){ if(l==r) return; int mid=(l+r)>>1; cdq(l, mid); cdq(mid+1, r); int p1, p2; p1=p2=mid+1; for(int i=l; i<=mid; i++){ while(p2<=r && p[p2].q<=p[i].q+k){ upd(p[p2].x, 1); p2++; } while(p1<p2 && p[p1].q<p[i].q-k){ upd(p[p1].x, -1); p1++; } ans+=query(p[i].R)-query(p[i].L-1); } for(int i=p1; i<p2; i++) upd(p[i].x, -1); inplace_merge(p+l, p+mid+1, p+r+1, cmp2); } int main(){ int n, x, r, q; scanf("%d%d", &n, &k); for(int i=1; i<=n; i++){ scanf("%d%d%d", &p[i].x, &p[i].r, &p[i].q); val[++cnt]=p[i].x; } sort(val+1, val+1+cnt); cnt=unique(val+1, val+1+cnt)-val-1; for(int i=1; i<=n; i++){ p[i].L=lower_bound(val+1, val+1+cnt, p[i].x-p[i].r)-val; p[i].R=upper_bound(val+1, val+1+cnt, p[i].x+p[i].r)-val-1; p[i].x=lower_bound(val+1, val+1+cnt, p[i].x)-val; } sort(p+1, p+1+n, cmp1); cdq(1, n); printf("%I64d\n", ans); return 0; }