[BZOJ2683]簡單題
阿新 • • 發佈:2018-08-11
algorithm 子矩陣 void 矩陣 ++ != ktr include tint
[BZOJ2683]簡單題
題目大意:
一個\(n\times n(n\le5\times10^5)\)的矩陣,初始時每個格子裏的數全為\(0\)。\(m(m\le2\times10^5)\)次操作,操作包含以下兩種:
- 將某個格子加上一個數;
- 詢問某個子矩陣的值。
思路:
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; }
[BZOJ2683]簡單題