BZOJ4170: 极光【曼哈顿转切比雪夫距离+离线二维数点(排序+cdq分治+树状数组)】

题目描述：

天空中出现了许多的北极光，这些北极光组成了一个长度为n的正整数数列a[i],远古之魔书上记载到：2个位置的graze值为两者位置差与数值差的和：
graze(x,y)=|x-y|+|a[x]-a[y]|。
要想破解天罚，就必须支持2种操作（k都是正整数）：
Modify x k：将第x个数的值修改为k。
Query x k：询问有几个i满足graze(x,i)<=k。
询问不仅要考虑当前数列，还要考虑任意历史版本，即统计任意位置上出现过的任意数值与当前的a[x]的graze值<=k的对数。（某位置多次修改为同样的数值，按多次统计）
n<=60000,操作数<=60000,Max{a[i]}<=100000

题目分析：

graze的限制相当于是个菱形，不好搞，神犇教我们旋转坐标轴：
$(x,y)=r(cos\theta ,sin\theta )$
$r(cos(\theta +\frac{\pi }{2}),sin(\theta +\frac{\pi }{2}))=(xcos\frac{\pi }{2}-ysin\frac{\pi }{2},ycos\frac{\pi }{2}+xsin\frac{\pi }{2})$
乘上一个$\sqrt{2}$就是$(x-y,y+x)$。
原来的限制距离是k，转为正方形边长是$\sqrt{2}k$，乘个$\sqrt{2}$就是2k。

所以相当于求转换坐标后在边长为2k的正方形内的点的数量。

可以在线二维线段树，树状数组套主席树，或者广为人知的离线cdq。
然而我太菜了从来没有打过树套树。

矩形可以分成四个小矩形相加减，就是三维偏序问题，两维坐标，一维操作时间。我是按x排序，按时间分治，然后y用树状数组。按时间分治有个好处就是可以方便地区分操作（不会有两个操作在同一时间），在cdq分治中不用再归并排序。

Code：

#include<cstdio>
#include<cctype>
#include<algorithm>
#define maxn 60005
using namespace std;
char cb[1<<18],*cs,*ct;
#define getc() (cs==ct&&(ct=(cs=cb)+fread(cb,1,1<<18,stdin),cs==ct)?0:*cs++)
inline void read(int &a){
    char c;while(!isdigit(c=getc()));
    for(a=c-'0';isdigit(c=getc());a=a*10+c-'0');
}
const int O = 100000, N = 160000;
int n,m,Q,a[maxn],ans[maxn];
int arr[N+5];
void upd(int i,int d){for(;i<=N;i+=i&-i) arr[i]+=d;}
int qsum(int i){int s=0;for(;i;i-=i&-i) s+=arr[i];return s;}
struct node{
    int x,y,v,t,id; node(){}
    node(int x,int y,int v,int t,int id):x(x),y(y),v(v),t(t),id(id){}
    bool operator < (const node &p)const{return x==p.x?t<p.t:x<p.x;}
}q[maxn*6],tmp[maxn*6];
void solve(int l,int r){
    if(l>=r) return;
    int mid=(l+r)>>1,lp=l,rp=mid+1;
    for(int i=l;i<=r;i++)
        if(q[i].t<=mid){
            tmp[lp++]=q[i];
            if(!q[i].id) upd(q[i].y,1);
        }
        else{
            tmp[rp++]=q[i];
            if(q[i].id) ans[q[i].id]+=q[i].v*qsum(q[i].y);
        }
    for(int i=l;i<=r;i++) if(q[i].t<=mid&&!q[i].id) upd(q[i].y,-1);
    for(int i=l;i<=r;i++) q[i]=tmp[i];
    solve(l,mid),solve(mid+1,r);
}
int main()
{
    int x,k; char op;
    read(n),read(m);
    for(int i=1;i<=n;i++) read(a[i]),q[i]=node(i-a[i]+O,i+a[i],0,i,0);
    int cnt=n;
    for(int i=1;i<=m;i++){
        while(!isalpha(op=getc()));read(x),read(k);
        if(op=='M') a[x]=k,q[++cnt]=node(x-a[x]+O,x+a[x],0,cnt,0);
        else{
            Q++; 
            if(x-a[x]+O-k-1>0){
                if(x+a[x]-k-1>0) q[++cnt]=node(x-a[x]+O-k-1,x+a[x]-k-1,1,cnt,Q);
                q[++cnt]=node(x-a[x]+O-k-1,min(x+a[x]+k,N),-1,cnt,Q);
            }
            if(x+a[x]-k-1>0) q[++cnt]=node(min(x-a[x]+O+k,N),x+a[x]-k-1,-1,cnt,Q);
            q[++cnt]=node(min(x-a[x]+O+k,N),min(x+a[x]+k,N),1,cnt,Q);
        }
    }
    sort(q+1,q+1+cnt);
    solve(1,cnt);
    for(int i=1;i<=Q;i++) printf("%d\n",ans[i]);
}