线段树HDU——1166（单点增减、区间求和）

应用：区间最值查询问题、区间求和

            连续区间的修改或单点更新

例题：HDU——1166敌兵布阵

//线段树功能:add:单点增减 sum:区间求和
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
#include<queue>
#include<math.h>
using namespace std;
#define maxn 2010
#define INF 99999999
#define MAX 50000
struct stu{
    int l,r,num;
}p[3*MAX];
int n;
void build(int i,int a,int b)//构造以【a，b】为根的线段树
{//当前考察节点为i
    int m=(a+b)/2;//把a，b所属的区间一分为二
    p[i].num=0;
    p[i].l=a;
    p[i].r=b;
    if (a!=b){
        if (a<=m)
            build(2*i,a,m);//以【a，m】为根继续构造
        if (b>m)
            build(2*i+1,m+1,b);//以【m+1，b】为根继续构造
    }
}

int sum(int i,int a,int b)//第a个营地到第b个营地的总人数
{
    int m=(p[i].l+p[i].r)/2;

    if (p[i].l==a && p[i].r==b)
        return p[i].num;
    else if (b<=m)//如果要查询的右区间小于m，则要查找的区间在左子树上
        return sum(i*2,a,b);
    else if (a>m)
        return sum(i*2+1,a,b);
    else
        return sum(i*2,a,m)+sum(i*2+1,m+1,b);
}

void add(int i,int a,int val)//将第a个营地的人数+val
{
    int m=(p[i].l+p[i].r)/2;

    if (p[i].l==p[i].r)//如果递归到子节点（即a点），将a点的num值加val
        p[i].num+=val;
    else if (a<=m)//a<=m表示左子树的区间内包含a点
    {
            p[i].num+=val;//将包含a点的区间num值加val
            add(i*2,a,val);//以此节点继续递归
    }
    else
    {
        p[i].num+=val;//右子树的区间内包含a点
        add(i*2+1,a,val);
    }
}

int main()
{
    int i,j,t,a,b,val;
    char ch[101];

    scanf("%d",&t);
    for(j=1;j<=t;j++)
    {
        scanf("%d",&n);
        build(1,1,n);//构造由1到n的线段树
        for (i=1;i<=n;i++)
        {
            scanf("%d",&val);
            add(1,i,val);
        }
        printf("Case %d:\n",j);
        while (scanf("%s",ch) && strcmp(ch,"End")!=0)
        {
            scanf("%d%d",&a,&b);
            if (strcmp(ch,"Add")==0)
                add(1,a,b);
            else if (strcmp(ch,"Sub")==0)
                add(1,a,-b);
            else
                printf("%d\n",sum(1,a,b));
        }
    }
    return 0;
}