hdu 1892 See you~(树状数组求二维前缀和)

题意:
有一个二维的方格阵,每个方格有一本书,现在最其中一个矩形区间进行增删询问操作,输出对于结果。
思路:
树状数组求二维前缀和基础题。
注意他输入的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;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值