HDU - 1166 - 敌兵布阵 线段树的单点修改，区间求和

Posted 2021-02-06 bluefly-hrbust

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxx = 50030;
int arry[maxx];
struct node
{
    int left;
    int right;
    int mid;
    int val;
} a[maxx*4];
void pushup(int root)
{
    a[root].val=a[root<<1].val+a[root<<1|1].val;//把laze标记往上传递。不能是加等于，就是等于，因为底部被更新后，需要往上的是新的内容
}
void buildtree(int root,int l,int r)
{
    int mid=(l+r)/2;
    a[root].left=l;
    a[root].right=r;
    a[root].mid=mid;
    a[root].val=0;
    if(l==r)
    {
        a[root].val=arry[l];//到最低端的时候
        return;
    }
    buildtree(root*2,l,mid);
    buildtree(root*2+1,mid+1,r);
    pushup(root);//把这个区间创建完成后，需要把更新往上更新
}
void updata(int root,int tag,int val)//单点更新
{
    int l = a[root].left,r=a[root].right;
    if(l==r)
    {
        a[root].val+=val;
        return;
    }
    int mid=(l+r)/2;
    if(tag<=mid)updata(2*root,tag,val); 
    else updata(root*2+1,tag,val);
    pushup(root);
}
int query(int root,int ql,int qr)
{
    int res=0;
    int l=a[root].left,r=a[root].right;
    int mid=(l+r)/2;
    if(ql<=l && qr>=r)return a[root].val;//询问的区间是比这个区间大的，那么代表这个区间的内部的所有都要被选到，因此无需再往下遍历
    if(ql<=mid)res+=query(root*2,ql,qr);//询问的左区间比当前区间小，那么这个区间的左边有一部分取不到，因此需要继续往下找
    if(qr>mid)res+=query(root*2+1,ql,qr);//询问的右区间比当前的右区间大，那么这个区间的右边有一部分取不到，因此也需要往下找
    return res;//最后把这两边的内容传递回去即可
}
int main()
{
    int t;
    int n;
    scanf("%d",&t);
    char op[20];
    int pos,val;
    int l,r;
    int cas=0;
    while(t--)
    {
        scanf("%d",&n);
        printf("Case %d:
",++cas);
        for (int i=1; i<=n; i++)
        {
            scanf("%d",&arry[i]);
        }
        buildtree(1,1,n);
        while(~scanf("%s",op))
        {
            if(op[0]==‘E‘)break;
            if(op[0]==‘A‘)
            {
                scanf("%d%d",&pos,&val);
                updata(1,pos,val);
            }
            else if(op[0]==‘S‘)
            {
                scanf("%d%d",&pos,&val);
           //     cout<<"???"<<-val<<endl;
                updata(1,pos,-val);
            }
            else
            {
                scanf("%d%d",&l,&r);
                printf("%d
", query(1,l,r));
            }
        }

    }
    return 0;
}