[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
;
int lowbit(const int &x) const {
return x&-x;
}
public:
void modify(int p,const int &x) {
for(;p<=n;p+=lowbit(p)) val
+=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