HDU 1166 敌兵布阵(线段树:点更新,区间求和)

HDU 1166 敌兵布阵(线段树:点更新,区间求和)

http://acm.hdu.edu.cn/showproblem.php?pid=1166

题意：

给你n个整数，然后给你多条命令，每条命令如下格式：

(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 表示结束，这条命令在每组数据最后出现;

分析:

线段树基本应用。单点增加，区间查询。

直接按线段树的思想实现线段树的PushUp，build，PushDown(本题不用)，query，update操作即可。
线段树的有关知识可见刘汝佳的《算法竞赛入门经典：训练指南》

AC代码(新)：用C++提交

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=50000+5;

//线段树需要维护的信息
int sum[maxn*4];
#define lson i*2, l, m
#define rson i*2+1, m+1, r

//i节点收集子节点的统计结果
void PushUp(int i)
{
    sum[i]=sum[i*2]+sum[i*2+1];
}

//递归建立线段树
void build(int i,int l,int r)
{
    if(l==r)
    {
        scanf("%d",&sum[i]);
        return ;
    }

    int m=(l+r)/2;
    build(lson);
    build(rson);
    PushUp(i);//收集子节点的结果
}

//在当前区间[l, r]内查询区间[ql, qr]间的目标值
//且能执行这个函数的前提是：[l,r]与[ql,qr]的交集非空
//其实本函数返回的结果也是 它们交集的目标值
int query(int ql,int qr,int i,int l,int r)
{
    //目的区间包含当前区间
    if(ql<=l && r<=qr) return sum[i];

    int m=(l+r)/2;
    int res=0;
    if(ql<=m) res += query(ql,qr,lson);
    if(m<qr) res += query(ql,qr,rson);
    return res;
}

//update这个函数就有点定制的意味了
//本题是单点更新，所以是在区间[l,r]内使得第id数的值+val
//如果是区间更新，可以update的参数需要将id变为ql和qr
void update(int id,int val,int i,int l,int r)
{
    if(l==r)
    {
        sum[i] += val;
        return ;
    }

    int m=(l+r)/2;
    if(id<=m) update(id,val,lson);
    else update(id,val,rson);
    PushUp(i);//时刻记住维护i节点统计信息正确性
}


int main()
{
    int T;
    scanf("%d",&T);
    for(int kase=1; kase<=T; kase++)
    {
        printf("Case %d:\n",kase);

        int n;//节点总数
        scanf("%d",&n);
        build(1,1,n);//建立线段树

        char str[20];
        int u,v;
        while(scanf("%s",str)==1 && str[0]!='E')
        {
            scanf("%d%d",&u,&v);

            if(str[0]=='Q') printf("%d\n",query(u,v,1,1,n));
            else if(str[0]=='A') update(u,v,1,1,n);
            else update(u,-v,1,1,n);
        }
    }
    return 0;
}

AC代码:203ms，要用C++提交

<span style="font-size:18px;">#include<cstdio>
#include<cstring>
using namespace std;
#define lson i*2,l,m
#define rson i*2+1,m+1,r

const int MAXN=50000+10;
int sum[MAXN*4];
void PushUp(int i)
{
    sum[i]=sum[i*2]+sum[i*2+1];
}
void build(int i,int l,int r)
{
    if(l==r)
    {
        scanf("%d",&sum[i]);
        return ;
    }
    int m =(r+l)/2;
    build(lson);
    build(rson);
    PushUp(i);
}
int query(int ql,int qr,int i,int l,int r)
{
    if(ql<=l&&r<=qr)
        return sum[i];
    int m =(r+l)/2;
    int res=0;
    if(ql<=m) res += query(ql,qr,lson);
    if(qr>m)res += query(ql,qr,rson);
    return res;
}
void update(int p,int add,int i,int l,int r)
{
    if(l==r)
    {
        sum[i] +=add;
        return;
    }
    int m =(l+r)/2;
    if(p<=m)update(p,add,lson);
    else update(p,add,rson);
    PushUp(i);
}
int main()
{
    int T;
    scanf("%d",&T);
    for(int kase=1;kase<=T;kase++)
    {
        printf("Case %d:\n",kase);
        int n;
        scanf("%d",&n);
        //memset(sum,0,sizeof(sum));//切记要初始化
        build(1,1,n);//建立初始树
        char str[10];
        while(scanf("%s",str)==1&&str[0]!='E')
        {
            int x,y;
            scanf("%d%d",&x,&y);
            if(str[0]=='Q')printf("%d\n",query(x,y,1,1,n));
            else if(str[0]=='A')update(x,y,1,1,n);
            else update(x,-y,1,1,n);
        }
    }
    return 0;
}
</span>