bzoj4170: 极光（CDQ分治）

4170: 极光

Time Limit: 30 Sec
Memory Limit: 512 MB

Description

“若是万一琪露诺（俗称rhl）进行攻击，什么都好，冷静地回答她的问题来吸引她。对方表现出兴趣的话，那就慢
慢地反问。在她考虑答案的时候，趁机逃吧。就算是很简单的问题，她一定也答不上来。”
–《上古之魔书》
天空中出现了许多的北极光，这些北极光组成了一个长度为$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$。
由于从前的天罚被圣王lmc破解了，所以rhl改进了她的法术，询问不仅要考虑当前数列，还要考虑任意历史版本，即统计任意位置上出现过的任意数值与当前的$a[x]$的$graze$值<=$k$的对数。（某位置多次修改为同样的数值，按多次统计）

Input

第1行两个整数$n$,$q$。分别表示数列长度和操作数。
第2行$n$个正整数，代表初始数列。
第3~$q+2$行每行一个操作。
$N≤40000$, 修改操作数$≤60000$, 询问操作数$≤10000$, $Max{a[i]}$(含修改)$≤80000$

Output

对于每次询问操作，输出一个非负整数表示答案

Sample Input

3 5
2 4 3
Query 2 2
Modify 1 3
Query 2 2
Modify 1 2
Query 1 1

Sample Output

2
3
3

解：

其实题面本来可以写得清楚易懂，不知道出题人的语文是怎么学的。其实就是加数，把时间、数轴、数组分别看做一维，可以发现要求的两维是一个切比雪夫距离。就是求一个菱形里的点数。当然可以使用KDtree。但是自然菱形是非常不好操作的，我们考虑把它转化成矩形，是不是就很好做了？
具体做法可以参见这个博客。
简单来说就是$(\frac {x+y}2,\frac {x−y}2)$
剩下的就大同小异了。
值得学习的就是曼哈顿距离和切比雪夫距离的相互转化。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define lowbit(x) (x&(-x))
using namespace std;
bool key;
struct lxy{
    int x,y,type,ans,tim,k;
    bool operator < (const lxy &QAQ)const{
      if(key==0){
        if(x!=QAQ.x) return x<QAQ.x;
        return type<QAQ.type;
      }
      return tim<QAQ.tim;
    }
}q[300005];

int data[300005];
int n,m,cnt,x,y;
int const p=500000;
int const N=2000000;
char s[20];
int b[2000005];

void modify(int x,int w)
{
    while(x<=N)
    {
        b[x]+=w;
        x=x+lowbit(x);
    }
}

int query(int x)
{
    int ret=0;
    while(x>0)
    {
        ret+=b[x];
        x=x-lowbit(x);
    }
    return ret;
}

void cdq(int l,int r)
{
    if(l==r) return;
    if(r<=n){
        key=0;sort(q+l,q+r+1);
        return;
    }
    int mid=(l+r)>>1;
    cdq(l,mid);cdq(mid+1,r);
    key=0;sort(q+l,q+r+1);
    for(int i=l;i<=r;i++)
    {
        if(q[i].tim<=mid&&q[i].type==0) modify(q[i].y,1);
        if(q[i].tim>mid&&q[i].type!=0) q[i].ans+=query(min(q[i].y+q[i].k,N))-query(max(1,q[i].y-q[i].k-1));
    }
    for(int i=l;i<=r;i++)
      if(q[i].type==0&&q[i].tim<=mid)
        modify(q[i].y,-1);
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)
    {
        ++cnt;q[cnt].tim=cnt;
        scanf("%d",&data[i]);
        q[cnt].x=i-data[i]+p;
        q[cnt].y=i+data[i]+p;
    }
    for(int i=1;i<=m;++i)
    {
        ++cnt;q[cnt].tim=cnt;
        scanf("%s",s);
        scanf("%d%d",&x,&y);
        if(s[0]=='Q'){
            q[cnt].type=2;
            q[cnt].x=max(1,x-data[x]+p-y-1);
            q[cnt].y=x+data[x]+p;
            q[cnt].k=y;
            cnt++;q[cnt].tim=cnt;
            q[cnt].type=1;
            q[cnt].x=min(N,x-data[x]+p+y);
            q[cnt].y=x+data[x]+p;
            q[cnt].k=y;
            continue;
        }
        data[x]=y;
        q[cnt].x=x-y+p;
        q[cnt].y=x+y+p;
    }
    cdq(1,cnt);
    key=1;sort(q+1,q+1+cnt);
    for(int i=1;i<=cnt;i++)
      if(q[i].type==2)
        printf("%d\n",q[i+1].ans-q[i].ans);
}