HDU 1892 二维树状数组模板题

最新推荐文章于 2023-06-15 21:06:27 发布

德鸭

最新推荐文章于 2023-06-15 21:06:27 发布

阅读量395

点赞数

CC 4.0 BY-SA版权

分类专栏：线段树

本文链接：https://blog.youkuaiyun.com/qq_36908995/article/details/77587442

线段树专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种使用二维稀疏数组实现高效查询与更新操作的方法，通过具体实例展示了如何进行区域求和、单点赋值及删除等操作，并提供了一个完整的C++代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

假设每个格子代表一个数A[i][j]，i是横坐标，j是纵坐标，左上角的坐标为（1，1)我们要求红色区域元素之和，设sum(i,j)表示以i,j坐标为右下角坐标，以0,0为左上角坐标矩形内的元素之和，

sum(c,d):绿色+黄色+红色+蓝色

sum(c,b-1):绿色+蓝色

sum(a-1,d):绿色+黄色

sum(a-1,b-1):绿色

则红色区域元素之和=sum(c,d)-sum(c,b-1)-sum(a-1,d)+sum(a-1,b-1)

画个图就可以得到。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1105;
int bit[maxn][maxn];

void updata(int x,int y,int val)
{
    for(int i=x;i<maxn;i+=(i &(-i)))
    {
        for(int j=y;j<maxn ; j+=(j & (-j)))
        {
            bit[i][j] += val;
        }
    }
}
int query(int x,int y)
{
    int ans=0;
    for(int i=x; i>0 ;i-=(i&(-i)))
    {
        for(int j=y;j>0;j-=(j&(-j)))
        {
            ans+=bit[i][j];
        }
    }
    return ans;
}
int main()
{
    int x1,x2,y1,y2,T,kiss=0,n,val;
    scanf("%d",&T);
    char s[30];
    while(T--)
    {
        for(int i=1;i<maxn;++i)
        {
            for(int j=1;j<maxn;++j)
            {
                bit[i][j] = 0 ;
            }
        }
        scanf("%d",&n);
        printf("Case %d:\n",++kiss);


        for(int i=1;i<maxn;++i)
        {
            for(int j=1;j<maxn ;++j)
            {
                updata(i,j,1);
            }
        }


        while(n--)
        {
            scanf("%s",s);
            if(s[0]=='S')
            {
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                x1++; x2++; y1++; y2++;

                if(x1<x2) swap(x1,x2);
                if(y1<y2) swap(y1,y2);
                int sum = query(x1,y1) - query(x2-1,y1) - query(x1,y2-1) + query(x2-1,y2-1);
                printf("%d\n",sum);
            }
            else if(s[0]=='A')
            {
                scanf("%d%d%d",&x1,&y1,&val);
                x1++; y1++;
                updata(x1,y1,val);
            }
            else if(s[0]=='D')
            {
                scanf("%d%d%d",&x1,&y1,&val);
                x1++; y1++;
                int sum = query(x1,y1) - query(x1-1,y1) - query(x1,y1-1) + query(x1-1,y1-1);
                int temp = min(sum,val);
                updata(x1,y1,-temp);
            }
            else
            {
                scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&val);
                x1++; y1++; x2++; y2++;
                int sum = query(x1,y1) - query(x1-1,y1) - query(x1,y1-1) + query(x1-1,y1-1);
                int temp = min(sum,val);
                updata(x1,y1,-temp);
                updata(x2,y2,temp);
            }
        }

    }
    return 0;
}