HDU 2642 树状数组

Posted 2020-08-01 Dan__ge

点击打开链接

题意：给个二维矩阵，矩阵有0或者1两个值，然后有三个操作，Q问区间和，剩下两个是更新点的值

思路：更新点的值直接更新就行了，然后询问区间和的时候就处理一下，每次问的是X1，Y1到X2，Y2的区间和，而树状数组的和是从1，1开始的，所以总的减去多于的在加上多减去的就OK了

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=1010;
int sum[maxn][maxn],n;
bool vis[maxn][maxn];
int lowbit(int x){return x&(-x);}
void update(int x,int y,int val){
    for(int i=x;i<maxn;i+=lowbit(i)){
        for(int j=y;j<maxn;j+=lowbit(j)){
                sum[i][j]+=val;
        }
    }
}
int msum(int x,int y){
    int res=0;
    for(int i=x;i>0;i-=lowbit(i)){
        for(int j=y;j>0;j-=lowbit(j)){
                res+=sum[i][j];
        }
    }
    return res;
}
int main(){
    int x1,x2,y1,y2;
    char ch[10];
    while(scanf("%d",&n)!=-1){
        memset(sum,0,sizeof(sum));
        memset(vis,0,sizeof(vis));
        while(n--){
            scanf("%s",ch);
            if(ch[0]=='B'){
                scanf("%d%d",&x1,&y1);
                x1++;y1++;
                if(vis[x1][y1]) continue;
                vis[x1][y1]=1;update(x1,y1,1);
            }else if(ch[0]=='D'){
                scanf("%d%d",&x1,&y1);
                x1++;y1++;
                if(vis[x1][y1]) update(x1,y1,-1);
                vis[x1][y1]=0;
            }else if(ch[0]=='Q'){
                scanf("%d%d%d%d",&x1,&x2,&y1,&y2);
                x1++;x2++;y1++;y2++;
                if(x1>x2) swap(x1,x2);
                if(y1>y2) swap(y1,y2);
                int ans=msum(x2,y2)-msum(x1-1,y2)-msum(x2,y1-1)+msum(x1-1,y1-1);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}