牛客算法5场-情人节的电灯泡-二维树状数组+区间查询，单点修改

情人节的电灯泡

这是一道二维树状数组的引用，和之前的一道有一点差别；

主要注意这里的查询操作；

 

　　　　　　getsum(x,y)+getsum(a-1,b-1)-getsum(x,b-1)-getsum(a-1,y)

 

#include <iostream>
using namespace std;
int n,m;
int sum[1005][1005],vis[1005][1005];
int lowbit(int x)
{
    return x & (-x);
}
void up(int x,int y,int c)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        for(int j=y;j<=n;j+=lowbit(j))
        {
            sum[i][j]+=c;
           
        }
    }
}
int getsum(int x,int y)
{
    int ans = 0;
    for(int i=x;i>0;i-=lowbit(i))
    {
        for(int j=y;j>0;j-=lowbit(j))
        {
            ans += sum[i][j];
        }
    }
    return ans;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            int x;
            cin>>vis[i][j];
            if(vis[i][j])up(i,j,1);
        }
    }
    for(int i=1;i<=m;i++)
    {
        int tag,a,b;
        cin>>tag>>a>>b;
        if(tag==1)
        {
            if(vis[a][b]){
                up(a,b,-1);
                vis[a][b]=0;
            }
            else
            {
                up(a,b,1);
                vis[a][b]=1;
            }
        }
        else
        {
            int x,y;
            cin>>x>>y;
            cout<<getsum(x,y)+getsum(a-1,b-1)-getsum(x,b-1)-getsum(a-1,y)<<endl;
        }
    }
    return 0;
}