题意:
有一个二维的方格阵,每个方格有一本书,现在最其中一个矩形区间进行增删询问操作,输出对于结果。
思路:
树状数组求二维前缀和基础题。
注意他输入的x1,y1有可能是0,所以++x1,++y1。
还有一点,因为lowbit(x)表示x所管理的区域,又因为这道题里每个格子里最开始有一本书,所以初始化时可以直接把二维数组a[i][j] = lowbit(i) * lowbit(j)
l。复杂度由n^2logn进化为n^2。
#include <bits/stdc++.h>
#define lowbit(x) (x&(-x))
using namespace std;
const int maxn=1100;
int tc,q,ft[maxn][maxn];
int getsum(int x, int y){
int sum=0;
for(int i=x; i>0; i-=lowbit(i)){
for(int j=y; j>0; j-=lowbit(j)){
sum+=ft[i][j];
}
}
return sum;
}
int update(int x, int y, int val){
for(int i=x; i<=1005; i+=lowbit(i)){
for(int j=y; j<=1005; j+=lowbit(j)){
ft[i][j]+=val;
}
}
}
int query(int x1, int y1, int x2, int y2){
//cerr<<getsum(x2,y2)<<' '<<getsum(x1-1,y2)<<' '<<getsum(x1,y1-1)<<' '<<getsum(x1-1,y1-1)<<endl;
return (getsum(x2,y2)-getsum(x1-1,y2)-getsum(x2,y1-1)+getsum(x1-1,y1-1));
}
int main() {
ios::sync_with_stdio(false);cin.tie(0);cout.precision(10);cout << fixed;
#ifdef LOCAL_DEFINE
freopen("input.txt", "r", stdin);
#endif
int kase=1;
cin>>tc;
while(tc--){
cin>>q;
char op;
int x1,y1,x2,y2,n1;
memset(ft,0,sizeof(ft));
for(int i=1; i<=1005; ++i) for(int j=1; j<=1005; ++j) ft[i][j]=lowbit(i)*lowbit(j);
cout<<"Case "<<kase++<<":"<<'\n';
for(int i=0; i<q; ++i){
cin>>op;
if(op=='S'){
cin>>x1>>y1>>x2>>y2;
++x1; ++y1;
++x2; ++y2;
if(x1>x2) swap(x1,x2);
if(y1>y2) swap(y1,y2);
cout<<query(x1,y1,x2,y2)<<'\n';
} else if(op=='A'){
cin>>x1>>y1>>n1;
++x1; ++y1;
update(x1,y1,n1);
} else if(op=='D'){
cin>>x1>>y1>>n1;
++x1; ++y1;
int num=min(n1, query(x1,y1,x1,y1));
update(x1,y1,-num);
} else if(op=='M'){
cin>>x1>>y1>>x2>>y2>>n1;
++x1; ++y1; ++x2; ++y2;
int num=min(n1, query(x1,y1,x1,y1));
update(x1,y1,-num);
update(x2,y2,num);
}
}
}
#ifdef LOCAL_DEFINE
cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
return 0;
}