【cdq分治】[BOI2007]Mokia 摩基亚

【题目】

传送门

【题解】

• 看一遍题，咦，这不二维树状数组吗？
• 慢着，W<=2000000·····
• 这时我果断打开了标签：“分治”？
• 难道是cdq？
• 再仔细看题，先按过去做二维树状数组的方法，将矩阵求和转化为求四个前缀子矩阵和
• 而在t时间求前缀子矩阵$(x,y)$的和，只用考虑时间在t以前，修改位置$x1<=x,y1<=y$的修改吗？
• 于是，这道题变成了cdq模板——三维偏序

【代码】

#include<bits/stdc++.h>
using namespace std;
#define int long long //博主太懒了，大家千万不要向他学习
inline int read(){
	int x=0,f=1;char ch=getchar();
	while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
	while(isdigit(ch)){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
	return x*f;
}
const int N=2e6+10;
const int M=160010;
int c[N],s,w,num,ans[M];
inline int lowbit(int x){
	return x&(-x);
}
inline void add(int x,int v){
	while(x<=w){
		c[x]+=v;
		x+=lowbit(x);
	}
}
inline int query(int x){
	int sum=0;
	while(x){
		sum+=c[x];
		x-=lowbit(x);
	}
	return sum;
}
struct node{
	int f,t,x,y,v;
}a[M<<1];
int n;
inline bool cmp1(node a,node b){
	if(a.t==b.t&&a.x==b.x) return a.y<b.y;
	if(a.t==b.t) return a.x<b.x;
	return a.t<b.t;
}
inline bool cmp2(node a,node b){
	if(a.x==b.x) return a.y<b.y;
	return a.x<b.x;
}
inline void cdq(int l,int r){
	if(l>=r) return ;
	int mid=(l+r)>>1;
	cdq(l,mid);cdq(mid+1,r);
	sort(a+l,a+mid+1,cmp2);sort(a+mid+1,a+r+1,cmp2);
	int i=l,j=mid+1;
	for(;j<=r;++j){
		while(a[i].x<=a[j].x&&i<=mid){
			if(a[i].f==1) add(a[i].y,a[i].v);
			i++;     
		}
		if(a[j].f==20)ans[a[j].t]-=query(a[j].y);
		if(a[j].f==21)ans[a[j].t]+=query(a[j].y);
	}
	for(int k=l;k<i;++k) if(a[k].f==1)add(a[k].y,-a[k].v);
} //cdq模板不在话下
signed main(){
	s=read();w=read();
	while(1){
		int f=read();
		if(f==3) break; 
		if(f==1){
			a[++n].f=1;a[n].t=n;
			a[n].x=read();a[n].y=read();a[n].v=read();
		}
		else{
			num++;
			int x1=read(),y1=read(),x2=read(),y2=read();
			a[++n].f=21;a[n].x=x1-1;a[n].y=y1-1;a[n].t=num;
			a[++n].f=20;a[n].x=x1-1;a[n].y=y2;a[n].t=num;
			a[++n].f=20;a[n].x=x2;a[n].y=y1-1;a[n].t=num;
			a[++n].f=21;a[n].x=x2;a[n].y=y2;a[n].t=num;
			ans[num]=s*(x2-x1+1)*(y2-y1+1); //将矩阵求和转化为求四个前缀子矩阵和
		}
	}
	cdq(1,n);
	for(int i=1;i<=num;++i)
		printf("%lld\n",ans[i]);
	return 0;
}