前段时间遇到线段树过不了,树状数组却过了的题。(其实线段树过得了的)
回忆了下树状数组。
主要原理,还是二进制位数,每一项的和表示其为它的前((最后一位1及其后)的二进制数)和,可从二进制图来看。(用线段树想一想其实只是线段树编号不同而已,本质类似)
写了下二维树状数组,几乎和一维相同,也没必要不同。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
#include <cstdio> #include <cstring> int l,r,x,y,n,a,p,sum[1125][1125]; inline int lowbit(int x) { return x&(-x); } void update(int x, int y, int p) { for(int i=x; i<=n; i+=lowbit(i)) for(int j=y; j<=n; j+=lowbit(j)) sum[i][j]+=p; } int query(int x, int y) { int ret=0; for(int i=x; i>0; i-=lowbit(i)) for(int j=y; j>0; j-=lowbit(j)) ret+=sum[i][j]; return ret; } int main() { while(scanf("%d",&p)!=EOF) { if(p==3) break; switch (p) { case 0: memset(sum,0,sizeof(sum)); scanf("%d",&n); break; case 1: scanf("%d%d%d",&x,&y,&a); update(x+1,y+1,a); break; case 2: scanf("%d%d%d%d",&x,&y,&l,&r); printf("%d ",query(l+1,r+1)-query(l+1,y)-query(x,r+1)+query(x,y)); break; } } return 0; }