hdu 1166 敌兵布阵

最新推荐文章于 2019-01-24 21:53:19 发布

readlnh

最新推荐文章于 2019-01-24 21:53:19 发布

阅读量377

点赞数

CC 4.0 BY-SA版权

分类专栏：线段树初级文章标签：线段树

本文链接：https://blog.youkuaiyun.com/readlnh/article/details/46917871

线段树初级专栏收录该内容

4 篇文章

订阅专栏

本文通过一个具体的实战案例，详细解析了如何使用线段树解决一类典型的问题：针对敌方工兵营地的人数变动进行高效的单点更新及区间查询操作。文章提供了完整的代码实现，并附带详细的注释说明。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意： 敌人有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 表示结束，这条命令在每组数据最后出现;

这题显而易见，是典型的线段树，Add和Sub命令本质上是同一个命令，即线段树的单点更新，Q命令就是求i到j这个区间的和。没什么坑点。

代码:

#include <cstdio>
#include <cstring>

int sum[50000 << 2];
int n;

void build(int rt,int l, int r)
{
    if(l == r)
    {
        scanf("%d", &sum[rt]);
        //printf("%d %d %d\n",l,rt,sum[rt]);
        return;
    }
    build(rt <<1 , l, (l+r)>> 1);
    build((rt << 1) + 1, ((l+r) >> 1) + 1 , r);
    sum[rt] = sum[rt << 1] + sum[(rt << 1) + 1];
    //printf("%d %d\n", rt, sum[rt]);
}

void update(int p, int add, int l, int r, int rt)
{
    //printf("%d %d\n", l, r);
    if(l == r)
    {
        //printf("%d\n",l);
        sum[rt] += add;
        return;
    }
    int mid = (l + r) >> 1;
    //printf("%d\n",mid);
    if (p <= mid)
        update(p, add , l, mid, rt << 1);
    else
        update(p, add , mid + 1, r, (rt << 1) + 1);
    sum[rt] = sum[rt << 1] + sum[(rt << 1) +1];
}

int query(int L, int R, int l, int r, int rt)
{
    //printf("%d %d\n",l,r);
    if(L <= l && R >= r)  //当前区间在所求区间内
    {
        //printf("%d\n",sum[rt]);
        return (sum[rt]);
    }
    int mid = (l + r) >> 1;
    int ret = 0;
    if(L <= mid)
        ret += query(L, R, l, mid, rt << 1);
    //printf("%d %d %d\n",l,r,ret);
    if(R > mid)
        ret += query(L, R, mid+1, r, (rt << 1)+1);
    //printf("%d %d %d\n",l,r,ret);
    return ret;
}

int main()
{
    int T;
    scanf("%d",&T);
    for(int i=1;i<=T;i++)
    {
        printf("Case %d:\n",i);
        scanf("%d", &n);
        build(1,1,n);
        char s[10];
        while(scanf("%s",s),s[0]!='E')
        {
            int a,b;
            scanf("%d %d",&a ,&b);
            if(s[0] == 'Q')
            {
                printf("%d\n",query(a, b, 1, n, 1));
            }
            else if(s[0] == 'A')
            {
                update(a, b, 1, n, 1);
            }
            else if(s[0] == 'S')
                update(a, -b, 1, n, 1);
        }
    }
    return 0;
}