BZOJ4170: 极光

BZOJ4170

对于$graze(x,y)=|x-y|+|a[x]-a[y]|<=k$
将位置看做$x$坐标，$a$值看做$y$坐标。就是求和某个点的曼哈顿距离小于等于$k$的点数个数。
发现这个范围是一个菱形。将坐标系旋转一下。$(x,y)$变成$(x-y,x+y)$，从曼哈顿距离变成切比雪夫距离。也就变成了求某个正方形内点的个数。（以$（x-y,x+y）$为中心，边长为$2k$）
发现这就是个很$nice$的$CDQ$分治的模型（三维偏序，$CDQ$+树状数组）。修改某点就变成了加值。很完美！
有一点要注意的就是询问的$y$坐标可能小于等于0。这种情况直接不管就好了，就不加入操作中。因为加值操作中所有的$y$都大于0.所以这些小于等于0的询问是废掉的。然后就是正常上板子了。

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <cmath>
#include <algorithm>
#define N 300005
#define INF 0x7fffffff
using namespace std;
typedef pair<int,int> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,tot,num,Mx,Id;
int szsz[N],ans[N],a[N];

int lowbit(int x) {
    return x&-x;
}

class Ope {
    public:
        int type,cnt,id,x,y,w;
    Ope(){}
    Ope(int tt,int cc,int ii,int xx,int yy,int ww) {
        type=tt,cnt=cc,id=ii,x=xx,y=yy,w=ww;
    }
}Q[N],tmp[N];

bool operator <(Ope a,Ope b) {
    return a.x<b.x||(a.x==b.x&&a.type<b.type);
}

void Sumup(int x,int y) {
    //printf("%d %d\n",x,y);
    for(int i=x;i<=Mx;i+=lowbit(i)) szsz[i]+=y;
}

int query(int x) {
    int rtn=0;
    //printf("%d ",x);
    for(int i=x;i;i-=lowbit(i)) rtn+=szsz[i];
    //printf("%d\n",rtn);
    return rtn;
}

void Clear(int x) {
    for(int i=x;i;i-=lowbit(i)) szsz[i]=0;
}

void CDQ(int L,int R)
{
    if(L==R) return;
    int mid=L+R>>1;
    CDQ(L,mid);CDQ(mid+1,R);
    int p=L,q=mid+1,o=0;
    while(p<=mid&&q<=R) {
        if(Q[p]<Q[q]) {
            if(!Q[p].type) Sumup(Q[p].y,1);
            tmp[o++]=Q[p++];
        }
        else {
            if(Q[q].type) 
                ans[Q[q].cnt]+=query(Q[q].y)*Q[q].w;
            tmp[o++]=Q[q++];
        }
    }
    while(p<=mid) 
    {
        if(!Q[p].type) Sumup(Q[p].y,1);
        tmp[o++]=Q[p++];
    }
    while(q<=R) {
        if(Q[q].type) 
            ans[Q[q].cnt]+=query(Q[q].y)*Q[q].w;
        tmp[o++]=Q[q++];
    }
    for(int i=0;i<o;i++)
    {
        if(!tmp[i].type&&tmp[i].id<=mid)
            Sumup(tmp[i].y,-1);
        Q[i+L]=tmp[i];
    }
    //printf("\n");
}

void Add(int x,int y,int k)
{
    int xx=x-y,yy=x+y;
    int x1=xx-k-1,y1=yy-k-1,x2=xx+k,y2=yy+k;
    Mx=max(Mx,y2);num++;
    if(y1>0) Q[++tot]=Ope(1,num,++Id,x1,y1,1);
    Q[++tot]=Ope(1,num,++Id,x1,y2,-1);
    if(y1>0) Q[++tot]=Ope(1,num,++Id,x2,y1,-1);
    Q[++tot]=Ope(1,num,++Id,x2,y2,1);
}

int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++) a[i]=read(),Q[++tot]=Ope(0,0,++Id,i-a[i],i+a[i],0),Mx=max(Mx,i+a[i]);
    for(int i=1;i<=m;i++)
    {
        char ch[10];scanf("%s",ch);int x,y;
        x=read(),y=read();
        if(ch[0]=='M') a[x]=y,Q[++tot]=Ope(0,0,++Id,x-a[x],x+a[x],0),Mx=max(Mx,x+a[x]);
        else Add(x,a[x],y);
    }
    CDQ(1,tot);
    //for(int i=1;i<=tot;i++) printf("%d %d %d\n",Q[i].type,Q[i].x,Q[i].y);
    for(int i=1;i<=num;i++) printf("%d\n",ans[i]);

    return 0;
}