hdu1166（线段树单点更新、查询）

题目链接：http://hdu.hustoj.com/showproblem.php?pid=1166

题意：

　　第一行一个整数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条命令

思路：

　　典型的线段树单点更新+查询

代码：

#include <iostream>
using namespace std;

const int maxn = 50010;
int segTree[maxn * 4];  //1-n子区间数量不超过4*n


void pushUp(int root)   //更新父节点值
{
    segTree[root] = segTree[root * 2] + segTree[root * 2 + 1];  //这题的操作是求和
}

void build(int root, int left, int right)  //建立线段树
{
    if (left == right) //到了叶子节点就输入叶子结点的值
    {
        cin >> segTree[root];
        return;
    }

    int mid = (left + right) / 2;

    //递归建立左子树和右子树
    build(root * 2, left, mid);
    build(root * 2 + 1, mid + 1, right);

    //每次更新父节点值
    pushUp(root);
}

void update(int root, int p, int add, int left, int right) //单节点更新，p为待更新节点下标，add为需增加或减少的值
{
    if (left == right)  //找到单节点就更新
    {
        segTree[root] += add;
        return;
    }

    //二分查找指定节点
    int mid = (left + right) / 2;
    if (p <= mid)
    {
        update(root * 2, p, add, left, mid);
    }
    else
    {
        update(root * 2 + 1, p, add, mid + 1, right);
    }

    //每次更新父节点值
    pushUp(root);
}

int query(int root, int q_left, int q_right, int now_left, int now_right)  //查询区间
{
    if (q_left <= now_left && q_right >= now_right)  //当前节点区间包含在查询区间内
    {
        return segTree[root];
    }

    int mid = (now_left + now_right) / 2;
    int sum = 0;
    if (q_left <= mid)
    {
        sum += query(root * 2, q_left, q_right, now_left, mid);
    }
    if (q_right > mid)
    {
        sum += query(root * 2 + 1, q_left, q_right, mid + 1, now_right);
    }
    return sum;
}

int main()
{
    ios::sync_with_stdio(false);    //取消cin于stdin的同步，不加这句除非用scanf和printf，不然会TLE
    int t, n;
    char op[10];
    cin >> t;
    for (int i = 1; i <= t; i++)
    {
        cout << "Case " << i << ":" << endl;
        cin >> n;
        build(1, 1, n);
        while(cin >> op)
        {
            if (op[0] == 'E')
            {
                break;
            }
            int a, b;
            cin >> a >> b;
            if (op[0] == 'Q')   //询问
            {
                int ans = query(1, a, b, 1, n);
                cout << ans << endl;
            }
            else if (op[0] == 'S')  //减
            {
                update(1, a, -b, 1, n);
            }
            else if (op[0] == 'A')  //加
            {
                update(1, a, b, 1, n);
            }
        }
    }
    return 0;
}

体会：

　　线段树的入门题，加深对线段树的理解和熟悉一下相关操作的代码编写。