[BZOJ2683]简单题/[BZOJ1176][BalkanOI2007]Mokia

[BZOJ2683]简单题

题目大意：

一个\(n\times n(n\le5\times10^5)\)的矩阵，初始时每个格子里的数全为\(0\)。\(m(m\le2\times10^5)\)次操作，操作包含以下两种：

1. 将某个格子加上一个数；
2. 询问某个子矩阵的值。

思路：

CDQ分治+树状数组。

源代码：

#include<cstdio>
#include<cctype>
#include<algorithm>
inline int getint() {
	register char ch;
	while(!isdigit(ch=getchar()));
	register int x=ch^'0';
	while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
	return x;
}
const int M=8e5+1,C=2e5,N=5e5+1;
struct Query {
	int type,t,id,x,y,v;
};
Query a[M];
int n,ans[C];
inline bool cmp1 (const Query &p1,const Query &p2) {
	if(p1.t==p2.t) {
		if(p1.x==p2.x) return p1.y<p2.y;
		return p1.x<p2.x;
	}
	return p1.t<p2.t;
}
inline bool cmp2 (const Query &p1,const Query &p2) {
	if(p1.x==p2.x) return p1.y<p2.y;
	return p1.x<p2.x;
}
class FenwickTree {
	private:
		int val[N];
		int lowbit(const int &x) const {
			return x&-x;
		}
	public:
		void modify(int p,const int &x) {
			for(;p<=n;p+=lowbit(p)) val[p]+=x;
		}
		int query(int p) const {
			int ret=0;
			for(;p;p-=lowbit(p)) ret+=val[p];
			return ret;
		}
};
FenwickTree t;
void cdq(const int &b,const int &e) {
	if(b==e) return;
	const int mid=(b+e)>>1;
	cdq(b,mid);
	cdq(mid+1,e);
	int p=b,q=mid+1;
	for(;q<=e;q++) {
		if(a[q].type==1) continue;
		for(;p<=mid&&a[p].x<=a[q].x;p++) {
			if(a[p].type==1) t.modify(a[p].y,a[p].v);
		}
		ans[a[q].id]+=t.query(a[q].y)*a[q].v;
	}
	while(--p>=b) {
		if(a[p].type==1) t.modify(a[p].y,-a[p].v);
	}
	std::inplace_merge(&a[b],&a[mid]+1,&a[e]+1,cmp2);
}
int main() {
	n=getint();
	int m=0,cnt=-1;
	for(register int opt=getint(),i=0;opt!=3;opt=getint(),i++) {
		if(opt==1) {
			const int x=getint(),y=getint(),v=getint();
			a[++m]=(Query){1,i,cnt,x,y,v};
		}
		if(opt==2) {
			const int x1=getint(),y1=getint(),x2=getint(),y2=getint();
			cnt++;
			if(x1!=1&&y1!=1) a[++m]=(Query){2,i,cnt,x1-1,y1-1,1};
			if(x1!=1) a[++m]=(Query){2,i,cnt,x1-1,y2,-1};
			if(y1!=1) a[++m]=(Query){2,i,cnt,x2,y1-1,-1};
			a[++m]=(Query){2,i,cnt,x2,y2,1};
		}
	}
	std::sort(&a[1],&a[m]+1,cmp1);
	cdq(1,m);
	for(register int i=0;i<=cnt;i++) {
		printf("%d\n",ans[i]);
	}
	return 0;
}

[BZOJ1176][BalkanOI2007]Mokia

改一下输入格式和数据范围，没什么区别。