【完全版】线段树初学--HDU 1166 敌兵布阵

【完全版】线段树初学--HDU 1166 敌兵布阵
链接： click here~~
题意：
第一行一个整数T，表示有T组数据。
每组数据第一行一个正整数N（N<=50000）,表示敌人有N个工兵营地，接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人（1<=ai<=50）。
接下来每行有一条命令，命令有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 表示结束，这条命令在每组数据最后出现;
每组数据最多有40000条命令
Output
对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问，输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

Sample Input
1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End
Sample Output
Case 1: 6 33 59

【解题思路】

单点更新:最最基础的线段树,只更新叶子节点,然后把信息用PushUP(int r)这个函数更新上来

具体的可以参考学长推荐的大牛博客：http://notonlysuccess.me/?p=978

代码：

#include <iostream>
#include <stdio.h>
#include <map>
#include <queue>
#include <stack>
#include <math.h>
#include <set>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
using namespace std;
int dir[4][2]= {{1,0},{-1,0},{0,1},{0,-1}};
#define mem(a) memset(a,0,sizeof(a))
#define LL long long
#define Max(a,b) a>b?a:b;
#define Min(a,b) a<b?a:b;
const int M=40005;
#define lchild l,m,rt<<1
#define rchild m+1,r,rt<<1|1
int tree[M<<2];
void Pushup(int rt)   /*  对于区间求和*/
{
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void Pushup_max(int rt) /*对于区间求最大值*/
{
    tree[rt]=Max(tree[rt<<1],tree[rt<<1|1]);
}
void Pushup_min(int rt)/*对于区间求最小值*/
{
    tree[rt]=Min(tree[rt<<1],tree[rt<<1|1]);
}
void build(int l,int r,int rt)      /*建树*/
{
    if(l==r)
    {
        scanf("%d",&tree[rt]);
        return;
    }
    int m=(l+r)>>1;
    build(lchild);
    build(rchild);
    Pushup(rt);
}
void update(int p,int add,int l,int r,int rt)   // 单点更新
{
    if(l==r)
    {
        tree[rt]+=add;
        return ;
    }
    int m=(l+r)>>1;
    if(p<=m) update(p,add,lchild);
    else update(p,add,rchild);
    Pushup(rt);
}
int query(int L,int R,int l,int r,int rt)      //区间查询
{
    if(L<=l&&r<=R)
    {
        return tree[rt];
    }
    int m=(l+r)>>1;
    int sum=0;
    if(L<=m) sum+=query(L,R,lchild);
    if(R>m) sum+=query(L,R,rchild);
    return sum;
}
int scan()
{
    int res = 0, flag = 0;
    char ch;
    if((ch = getchar()) == '-') flag = 1;
    else if(ch >= '0' && ch <= '9') res = ch - '0';
    while((ch = getchar()) >= '0' && ch <= '9')
        res = res * 10 + (ch - '0');
    return flag ? -res : res;
}
int main()
{
    int t,n,m,i,j,tot=1;
    scanf("%d",&t);
    while(t--)
    {
        printf("Case %d:\n",tot++);
        n=scan();
        build(1,n,1);
        char op[10];
        while(scanf("%s",op))
        {
            if(op[0]=='E') break;
            int a,b;
            a=scan();
            b=scan();
            if(op[0]=='Q') printf("%d\n",query(a,b,1,n,1));
            else if(op[0]=='S') update(a,-b,1,n,1);
            else update(a,b,1,n,1);
        }
    }
    return 0;
}

线段树的第一题，到现在才学，实在是有点晚了，加油！