[bzoj] 1176 Mokia || CDQ分治

原题

给出W×W的矩阵（S没有用，题目有误），给出无限次操作，每次操作的含义为：

输入1:你需要把(x,y)(第x行第y列)的格子权值增加a

输入2:你需要求出以左下角为(x1,y1),右上角为(x2,y2)的矩阵内所有格子的权值和,并输出

输入3:表示输入结束

---

因为修改之间相互独立，所以可以用CDQ。

三个维度分别为时间，x轴，y轴。

简单的三维偏序即可。

#include<cstdio>
#include<algorithm>
#define N 200010
using namespace std;
struct hhh
{
    int x,y,w,pos,id,op;
    bool operator < (const hhh &b) const
	{
	    if (x==b.x && y==b.y) return op<b.op;
	    if (x==b.x) return y<b.y;
	    return x<b.x;
	}
}q[N],tmp[N];
int s,w,n,ans[N],m,op,f[10*N];

int read()
{
    int ans=0,fu=1;
    char j=getchar();
    for (;j<'0' || j>'9';j=getchar()) if (j=='-') fu=-1;
    for (;j>='0' && j<='9';j=getchar()) ans*=10,ans+=j-'0';
    return ans*fu;
}

void addquery()
{
    int lx=read(),ly=read(),rx=read(),ry=read();
    int pos=++ans[0];
    q[++m].pos=pos;q[m].id=m;q[m].x=lx-1;q[m].y=ly-1;q[m].w=1;q[m].op=1;
    q[++m].pos=pos;q[m].id=m;q[m].x=rx;q[m].y=ry;q[m].w=1;q[m].op=1;
    q[++m].pos=pos;q[m].id=m;q[m].x=lx-1;q[m].y=ry;q[m].w=-1;q[m].op=1;
    q[++m].pos=pos;q[m].id=m;q[m].x=rx;q[m].y=ly-1;q[m].w=-1;q[m].op=1;
}

void add(int x,int y)
{
    while (x<=w)
    {
	f[x]+=y;
	x+=x&-x;
    }
}

int query(int x)
{
    int ans=0;
    while (x)
    {
	ans+=f[x];
	x-=x&-x;
    }
    return ans;
}

void CDQ(int l,int r)
{
    if (l==r) return ;
    int mid=(l+r)>>1,i=l,j=mid+1;
    for (int k=l;k<=r;k++)
    {
	if (q[k].id<=mid && !q[k].op)
	    add(q[k].y,q[k].w);
	if (q[k].id>mid && q[k].op)
	    ans[q[k].pos]+=q[k].w*query(q[k].y);
    }
    for (int k=l;k<=r;k++)//清空树状数组
	if (q[k].id<=mid && !q[k].op) add(q[k].y,-q[k].w);
    for (int k=l;k<=r;k++)
	if (q[k].id<=mid) tmp[i++]=q[k];
	else tmp[j++]=q[k];
    for (int k=l;k<=r;k++)
	q[k]=tmp[k];
    CDQ(l,mid);
    CDQ(mid+1,r);
}

int main()
{
    s=read();
    w=read();
    while (1)
    {
	op=read();
	if (op==1)
	{
	    q[++m].x=read();
	    q[m].y=read();
	    q[m].w=read();
	    q[m].id=m;
	}
	else if (op==2)
	    addquery();
	else break;
    }
    sort(q+1,q+m+1);
    CDQ(1,m);
    for (int i=1;i<=ans[0];i++)
	printf("%d\n",ans[i]);
    return 0;
}