hdu 1166 敌兵布阵(线段树入门第一题)

题目：hdu1166 敌兵布阵

题意：有N个兵营，每个兵营都给出了人数ai（下标从1开始），有四种命令：

（1）”Add i j"，表示第i个营地增加j人。

（2）“Sub i j”，表示第i个营地减少j人。

（3）“Query i j"，查询第i个营地到第j个营地的总人数。

（4）”End“，表示命令结束。

这题用线段树来做，比暴力优化很多，注意用scanf会比用cin效率提高很多。

先建立一颗空树（初始化），再分别写更新结点函数和查询函数。

本题没有坑，是线段树最简单的入门题~欢迎各位巨巨来A~

网上关于线段树的介绍很多很多，再次就不对线段树介绍了，注意：线段树需要开4*maxn的数组噢！

#include<stdio.h>
struct linetree{
int r,l,sum;
}tree[200005];
int n;
int min(int a,int b){return a<b?a:b;}
int max(int a,int b){return a>b?a:b; }
void init(int l,int r,int p){/*初始化*/
    tree[p].sum=0;
    tree[p].l=l;
    tree[p].r=r;
    if(l==r) return ;
    init(l,(l+r)/2,p*2);
    init((l+r)/2+1,r,p*2+1);
}
void add(int fi,int cnt,int p){/*fi目标点,cnt是增加人数,p是当前节点,*/
    tree[p].sum+=cnt;
    if(tree[p].r==tree[p].l)
        return;
    if(fi<=(tree[p].r+tree[p].l)/2)
        add(fi,cnt,2*p);
    else
        add(fi,cnt,2*p+1);
};
int ask(int l,int r,int p){/*l左,r右,p当前节点*/
    int s=0,mid;
    if(tree[p].l==l&&tree[p].r==r)
        return tree[p].sum;
    mid=(tree[p].l+tree[p].r)/2;
    if(l<=mid)
        s+=ask(l,min(mid,r),2*p);
    if(r>mid)
        s+=ask(max(mid+1,l),r,2*p+1);
    return s;
}

int main(){
    int t,tt=1;
    scanf("%d",&t);
    while(t--){
        printf("Case %d:\n",tt++);
        scanf("%d",&n);
        init(1,n,1);
        int num;
        for(int i=0;i<n;i++){
            scanf("%d",&num);
            add(i+1,num,1);
        }
        char s[10];
        while(1){
            scanf("%s",s);
            if(s[0]=='E') break;
            int i,j;
            scanf("%d%d",&i,&j);
            if(s[0]=='A')
                add(i,j,1);
            else if(s[0]=='S')
                add(i,-j,1);
            else if(s[0]=='Q')
                printf("%d\n",ask(i,j,1));
        }
    }
    return 0;
}