[Luogu4390][BOI2007]Mokia 摩基亚

luogu

题意

支持平面内单点加一个值以及矩阵求和。

平面大小\(W\le2*10^6\)，修改操作\(\le1.6*10^5\)，查询操作\(\le10^4\)

sol

\(CDQ\)写一发。

把一个询问拆成四个点，类似二维前缀和的形式。这样对于每一个询问，相当于就是问满足\(x_i\le X,y_i\le Y\)的权值和。

树状数组维护即可。

code

#include<cstdio>
#include<algorithm>
using namespace std;
int gi()
{
	int x=0,w=1;char ch=getchar();
	while ((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
	if (ch=='-') w=0,ch=getchar();
	while (ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
	return w?x:-x;
}
const int N = 2e6+5;
struct node{
	int id,x,y,v,opt;
	bool operator < (const node &b) const
		{
			if (x!=b.x) return x<b.x;
			if (y!=b.y) return y<b.y;
			return !opt;
		}
}p[N],q[N];
int W,n,m,c[N],ans[N];
void modify(int k,int v){while(k<=W)c[k]+=v,k+=k&-k;}
int query(int k){int s=0;while(k)s+=c[k],k-=k&-k;return s;}
void CDQ(int l,int r)
{
	if (l==r) return;
	int mid=l+r>>1;CDQ(l,mid);CDQ(mid+1,r);
	int L=l,R=mid+1;
	for (int i=l;i<=r;++i)
		if (L<=mid&&(R>r||p[L]<p[R]))
		{
			q[i]=p[L++];
			if (!q[i].opt) modify(q[i].y,q[i].v);
		}
		else
		{
			q[i]=p[R++];
			if (q[i].opt) ans[q[i].v]+=query(q[i].y)*q[i].opt;
		}
	for (int i=l;i<=r;++i)
	{
		p[i]=q[i];
		if (p[i].id<=mid&&!p[i].opt) modify(q[i].y,-q[i].v);
	}
}
int main()
{
	while (233)
	{
		int opt=gi();
		if (opt==0) W=gi();
		if (opt==1)
		{
			int x=gi(),y=gi(),v=gi();
			p[++n]=(node){n,x,y,v,0};
		}
		if (opt==2)
		{
			int x1=gi(),y1=gi(),x2=gi(),y2=gi();
			++m;
			if (x1>1&&y1>1) p[++n]=(node){n,x1-1,y1-1,m,1};
			if (x1>1) p[++n]=(node){n,x1-1,y2,m,-1};
			if (y1>1) p[++n]=(node){n,x2,y1-1,m,-1};
			p[++n]=(node){n,x2,y2,m,1};
		}
		if (opt==3) break;
	}
	CDQ(1,n);
	for (int i=1;i<=m;++i) printf("%d\n",ans[i]);
	return 0;
}