HDU 1166 敌兵布阵（线段树：单点add，区间求和）

题意：

 

给你n个整数，4种操作：

(1) Add i j，i和j为正整数,表示第i个整数增加j（j不超过30）
(2)Sub i j，i和j为正整数,表示第i个整数减少j（j不超过30）;
(3)Query i j，i和j为正整数,i<=j，表示询问第i到第j个整数的和;
(4)End 表示结束，这条命令在每组数据最后出现;

 

 

 

题解：

      线段树模板题：单点add，区间求和。

 

 

 

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<string>
#include<cstring>
#include<vector>
#include<functional>
#include<utility>
#include<set>
#include<map>
#include<cmath>
#include<cctype>
#define INF 0x3f3f3f3f

using namespace std;

const int maxn=50000+10;

//线段树需要维护的信息
int sum[maxn*4];

//i节点收集子节点的统计结果
void pushup(int i)
{
    sum[i]=sum[i*2]+sum[i*2+1];
}

//递归建立线段树
void build(int i,int l,int r)
{
    if(l==r)
    {
        scanf("%d",&sum[i]);
        return;
    }

    int m=(l+r)/2;
    build(i*2,l,m);
    build(i*2+1,m+1,r);
    pushup(i);//收集子节点的结果。
}


//在当前区间[l,r]内查询区间[ql,qr]间的目标值
int query(int ql,int qr,int i,int l,int r)
{
    if(ql<=l&&qr>=r)//目的区间包含当前区间
        return sum[i];

    int m=(l+r)/2;
    int res=0;

    if(ql<=m)res+=query(ql,qr,2*i,l,m);
    if(m<qr)res+=query(ql,qr,2*i+1,m+1,r);
    return res;
}

void update(int id,int val,int i,int l,int r)
{
    if(l==r)
    {
        sum[i]+=val;
        return;
    }

    int m=(l+r)/2;
    if(id<=m)update(id,val,2*i,l,m);//左递归
    else update(id,val,2*i+1,m+1,r);//右递归
    pushup(i);
}

int main()
{
    int t;cin>>t;
    for(int kase=1;kase<=t;kase++)
    {
        printf("Case %d:\n",kase);

        int n;
        cin>>n;

        build(1,1,n);

        char s[20];
        int u,v;
        while(scanf("%s",s)==1&&s[0]!='E')
        {
            scanf("%d%d",&u,&v);

            if(s[0]=='Q')cout<<query(u,v,1,1,n)<<endl;
            else if(s[0]=='A')update(u,v,1,1,n);
            else update(u,-v,1,1,n);
        }
    }
}