线段树专题 A(单点更新)

最新推荐文章于 2022-04-11 01:13:29 发布

ALLACS

最新推荐文章于 2022-04-11 01:13:29 发布

阅读量217

点赞数 1

CC 4.0 BY-SA版权

文章标签：算法线段树

本文链接：https://blog.youkuaiyun.com/m0_37428263/article/details/78249735

本文介绍了一种使用树状数组实现区间更新与查询的方法。通过递归构造树状数组，并实现节点值的更新与区间求和等功能，能够高效处理数组中元素的更新与查询操作。文章还提供了一个具体的实现示例。

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

#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=50000+5;
int num[maxn];
char str[10];
struct Node
{
    int l,r;
    int nsum;
}Tree[maxn*3];
//自上到下建树,自下到上赋值
void build(int i,int l,int r)
{
    Tree[i].l=l;
    Tree[i].r=r;
    if(l==r)                                //递归出口
    {
        Tree[i].nsum=num[l];
        return;
    }
    int mid=(l+r)>>1;
    build(i<<1,l,mid);
    build((i<<1)+1,mid+1,r);
    Tree[i].nsum=Tree[i<<1].nsum+Tree[(i<<1)+1].nsum;
}
//自上到下的更新
void Add(int i,int t,int b)             //t为军营编号,b为变动
{
        Tree[i].nsum+=b;
        if(Tree[i].l==t&&Tree[i].r==t) return;
        int mid=(Tree[i].l+Tree[i].r)>>1;
        if(t<=mid) Add(i<<1,t,b);
        else Add((i<<1)+1,t,b);
}
//查询
int query(int i,int l,int r)
{
    if(l==Tree[i].l&&r==Tree[i].r) return Tree[i].nsum;
    int mid=(Tree[i].l+Tree[i].r)>>1;
    if(r<=mid) return query((i<<1),l,r);
    else if(l>mid) return query((i<<1)+1,l,r);
            else return query(i<<1,l,mid)+query((i<<1)+1,mid+1,r);
}
int main()
{
    int t;
    while(scanf("%d",&t)==1)
    {
        int n,kase=0;
        while(t--)
        {
            scanf("%d",&n);
            for(int i=1;i<=n;i++)
                 scanf("%d",&num[i]);
            build(1,1,n);
            printf("Case %d:\n",++kase);
            while(scanf("%s",&str)==1)
            {
                if(str[0]=='E') break;
                int a,b;
                scanf("%d%d",&a,&b);
                if(str[0]=='A') Add(1,a,b);
                else if(str[0]=='S') Add(1,a,-b);
                        else printf("%d\n",query(1,a,b));
            }
        }
    }
    return 0;
}