1. 程式人生 > >codeforces 1046_problem_A AI robots (cdq分治)

codeforces 1046_problem_A AI robots (cdq分治)

傳送門

題意:給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;
}