[arc065e]Manhattan Compass

题目大意

不想讲。

做法

可以转45度角变成切比雪夫距离，这样图像是正方形。
先用bfs找出所有可以踩到的点（可以对每行每列维护set，一个被bfs到的直接在两颗平衡树中删去）。
然后对每个能被踩到的点统计与其切比雪夫距离为d的点的个数（可以对每行每列维护vector并二分）。
最后答案除以2。

#include<cstdio>#include<algorithm>#include<cmath>#include<set>#include<map>#include<vector>#define fo(i,a,b) for(i=a;i<=b;i++)using namespace std;typedef long long ll;const int maxn=100000+10;const ll inf=40000000000;map<ll,int> row,col;struct dong{    int id,y;    ll x;    friend bool operator <(dong a,dong b){        return a.x<b.x||a.x==b.x&&a.id<b.id;    }} zlt;set<dong> ro[maxn],co[maxn];set<dong>::iterator it;vector<dong> rw[maxn],cl[maxn];vector<dong>::iterator ti;int x[maxn],y[maxn],dl[maxn];ll p[maxn],q[maxn];bool bz[maxn];int i,j,k,l,t,n,m,xx,yy,a,b,head,tail,now,tot,top;ll ans,d;void del(int id){    int xx=x[id],yy=y[id];    zlt.id=id;zlt.x=q[yy];    ro[xx].erase(ro[xx].find(zlt));    zlt.id=id;zlt.x=p[xx];    co[yy].erase(co[yy].find(zlt));}void ins(int x,ll l,ll r){    int y;    dong wxh;    while (1){        wxh.id=0;wxh.x=l;        it=ro[x].lower_bound(wxh);        if (it==ro[x].end()) break;        zlt=*it;        if (zlt.x>r) break;        y=zlt.id;        bz[y]=1;        dl[++tail]=y;        del(y);    }}void inse(int x,ll l,ll r){    int y;    dong wxh;    while (1){        wxh.id=0;wxh.x=l;        it=co[x].lower_bound(wxh);        if (it==co[x].end()) break;        zlt=*it;        if (zlt.x>r) break;        y=zlt.id;        bz[y]=1;        dl[++tail]=y;        del(y);    }}void ask(int x,ll l,ll r){    /*int t=lower_bound(rw[x],rw[x]+s,r+1)-rw[x];    t-=lower_bound(rw[x],rw[x]+s,l)-rw[x];*/    dong wxh;    wxh.id=0;wxh.x=r+1;    zlt=*(lower_bound(rw[x].begin(),rw[x].end(),wxh));    int t=zlt.y;    wxh.id=0;wxh.x=l;    zlt=*(lower_bound(rw[x].begin(),rw[x].end(),wxh));    t-=zlt.y;    ans+=(ll)t;}void query(int x,ll l,ll r){    /*int s=cl[x].size();    int t=lower_bound(cl[x],cl[x]+s,r+1)-cl[x];    t-=lower_bound(cl[x],cl[x]+s,l)-cl[x];    ans+=(ll)t;*/    dong wxh;    wxh.id=0;wxh.x=r+1;    zlt=*(lower_bound(cl[x].begin(),cl[x].end(),wxh));    int t=zlt.y;    wxh.id=0;wxh.x=l;    zlt=*(lower_bound(cl[x].begin(),cl[x].end(),wxh));    t-=zlt.y;    ans+=(ll)t;}int main(){    scanf("%d%d%d",&n,&a,&b);    fo(i,1,n){        scanf("%d%d",&j,&k);        p[i]=x[i]=j-k;        q[i]=y[i]=j+k;    }    sort(p+1,p+n+1);    tot=unique(p+1,p+n+1)-p-1;    fo(i,1,n) x[i]=lower_bound(p+1,p+tot+1,x[i])-p;    sort(q+1,q+n+1);    top=unique(q+1,q+n+1)-q-1;    fo(i,1,n) y[i]=lower_bound(q+1,q+top+1,y[i])-q;    fo(i,1,tot) row[p[i]]=i;    fo(i,1,top) col[q[i]]=i;    d=max(abs(p[x[a]]-p[x[b]]),abs(q[y[a]]-q[y[b]]));    fo(i,1,n){        zlt.id=i;zlt.x=q[y[i]];        ro[x[i]].insert(zlt);        rw[x[i]].push_back(zlt);        zlt.id=i;zlt.x=p[x[i]];        co[y[i]].insert(zlt);        cl[y[i]].push_back(zlt);        /*rw[x[i]].push_back(q[y[i]]);        cl[y[i]].push_back(p[x[i]]);*/    }    fo(i,1,tot){        sort(rw[i].begin(),rw[i].end());        zlt.id=0;zlt.x=inf;        rw[i].push_back(zlt);        t=rw[i].size();        fo(j,0,t-1) rw[i][j].y=j;    }    fo(i,1,top){        sort(cl[i].begin(),cl[i].end());        zlt.id=0;zlt.x=inf;        cl[i].push_back(zlt);        t=cl[i].size();        fo(j,0,t-1) cl[i][j].y=j;    }    head=0;    dl[tail=1]=a;    bz[a]=1;    del(a);    while (head<tail){        now=dl[++head];        xx=x[now];yy=y[now];        if (row[p[xx]-d]) ins(row[p[xx]-d],q[yy]-d,q[yy]+d);        if (row[p[xx]+d]) ins(row[p[xx]+d],q[yy]-d,q[yy]+d);        if (col[q[yy]-d]) inse(col[q[yy]-d],p[xx]-d,p[xx]+d);        if (col[q[yy]+d]) inse(col[q[yy]+d],p[xx]-d,p[xx]+d);    }    fo(i,1,n)        if (bz[i]){            xx=x[i];yy=y[i];            if (row[p[xx]-d]) ask(row[p[xx]-d],q[yy]-d+1,q[yy]+d);            if (col[q[yy]-d]) query(col[q[yy]-d],p[xx]-d,p[xx]+d-1);            if (row[p[xx]+d]) ask(row[p[xx]+d],q[yy]-d,q[yy]+d-1);            if (col[q[yy]+d]) query(col[q[yy]+d],p[xx]-d+1,p[xx]+d);        }    ans/=2;    printf("%lld\n",ans);}