矩陣樹狀陣列(矩陣加矩陣求和)
阿新 • • 發佈:2018-12-15
將矩陣分為很多由lowbit 組成的小矩陣 , 然後就跟樹狀陣列一樣維護了
求和的時候用矩陣字首和的思想(s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1])
單點修改時將(x1,y1)加 , (x2+1,y2) 減 , (x1,y2+1) 減 , (x2+1,y2+1) 加
複雜度是 O((logn)*(logm))
我們考慮如何矩陣加
我們考慮差分陣列
現在放一個矩陣修改 , 矩陣求和的程式碼
#include<bits/stdc++.h> #define N 2050 #define LL long long using namespace std; LL t1[N][N],t2[N][N],t3[N][N],t4[N][N]; int n,m; void Up(int x,int y,int val){ for(int i=x;i<=n;i+=i&-i) for(int j=y;j<=m;j+=j&-j){ t1[i][j] += (LL)val; t2[i][j] += (LL)val * x; t3[i][j] += (LL)val * y; t4[i][j] += (LL)val * x * y; } } LL Q(int x,int y){ LL ans = 0; for(int i=x;i;i-=i&-i) for(int j=y;j;j-=j&-j){ ans += (LL)t1[i][j] * (x+1) * (y+1); ans -= (LL)t2[i][j] * (y+1); ans -= (LL)t3[i][j] * (x+1); ans += (LL)t4[i][j]; } return ans; } int main(){ scanf("%d%d",&n,&m); int op,x1,y1,x2,y2; while(~scanf("%d%d%d%d%d",&op,&x1,&y1,&x2,&y2)){ if(op==1){ int val; scanf("%d",&val); Up(x1,y1,val); Up(x1,y2+1,-val); Up(x2+1,y1,-val); Up(x2+1,y2+1,val); } if(op==2){ LL ans = Q(x2,y2) - Q(x2,y1-1) - Q(x1-1,y2) + Q(x1-1,y1-1); printf("%lld",ans); } } return 0; }