[kuangbin带你飞]专题七线段树 A - 敌兵布阵（单点修改，区间求和）

最新推荐文章于 2020-03-12 17:29:07 发布

xmudl

最新推荐文章于 2020-03-12 17:29:07 发布

阅读量206

点赞数

本文链接：https://blog.csdn.net/q3818908/article/details/78126801

版权

https://vjudge.net/problem/UESTC-1057

线段树基础题（单点修改，区间求和）

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<iostream>
#define lid (id << 1)
#define rid (id << 1 | 1)
using namespace std;
const int MAXN=50005;
int a[MAXN];
struct Node
{
    int l,r;
    int sum,Max;
    void updata(int x)
    {
        sum+=x; //增加
      //  sum=x; //修改
        Max=sum;

    }
}tr[MAXN*4];
void push_up(int id)
{
    tr[id].sum = tr[lid].sum + tr[rid].sum;
    tr[id].Max = max(tr[lid].Max,tr[rid].Max);
}
void build(int id,int l,int r)
{
    tr[id].l=l;tr[id].r=r;
    if(l==r) tr[id].sum=tr[id].Max=a[l];
    else
    {
         int mid=(l+r)>>1;
         build(lid,l,mid);
         build(rid|1,mid+1,r);
         push_up(id);
    }
}
void updata(int id,int x,int v)
{
    if(tr[id].l==x&&tr[id].r==x)  tr[id].updata(v);
    else{
        int mid=(tr[id].l+tr[id].r)>>1;
        if(x<=mid)  updata(lid,x,v);
        else       updata(rid,x,v);
        push_up(id);
    }
}
int query_sum(int id,int l,int r)
{
    if(l==tr[id].l&&r==tr[id].r) return tr[id].sum;
    int mid=(tr[id].l+tr[id].r)>>1;
    if(r<=mid)  return query_sum(lid,l,r);
    else if(l>mid)   return query_sum(rid,l,r);
    else  return query_sum(lid,l,mid)+query_sum(rid,mid+1,r);

}
/*int query_max(int id,int l,int r)
{
    if(l==tr[id].l&&r==tr[id].r) return tr[id].Max;
    int mid=(tr[id].l+tr[id].r)>>1;
    if(r<=mid)  return query_max(lid,l,r);
    else if(l>mid)   return query_max(rid,l,r);
    else  return max(query_max(lid,l,mid),query_max(rid,mid+1,r));

} */
char str[10];
int main()
{
    int T,iCase=1,n;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)  scanf("%d",&a[i]);
        build(1,1,n);  printf("Case %d:\n",iCase++);
        while(scanf("%s",&str))
        {
            if(strcmp(str,"End")==0) break;
            int x,y;
            scanf("%d%d",&x,&y);
            if(strcmp(str,"Add")==0) updata(1,x,y);
            else if(strcmp(str,"Sub")==0)  updata(1,x,-y);
            else  printf("%d\n",query_sum(1,x,y));
        }
    }
    return 0;
}