[BZOJ2683]简单题

[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;
}