hdu1166(线段树)

题意:给出一列数,之后做四种操作中的一种

(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条命令

思路:先建立线段树,之后相应的进行点修改,query的时候递归分段求就行了

代码:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#define N 50005
using namespace std;
int n;
int p,q;
struct Node
{
    int sum,l,r;
};
Node node[N<<2];
int num[N];

void built(int t,int l,int r)
{
    node[t].l=l;
    node[t].r=r;
    if(l==r) {
        node[t].sum=num[l];
        return;
    }
    int mid=(l+r)>>1;
    built(t<<1,l,mid);
    built(t<<1|1,mid+1,r);
    node[t].sum=node[t<<1].sum+node[t<<1|1].sum;
}

void add(int t,int l,int r)
{
    int mid=(l+r)>>1;
    if(l==r) node[t].sum+=q;
    else{
        if(p<=mid) add(t<<1,l,mid);
        else add(t<<1|1,mid+1,r);
        node[t].sum=node[t<<1].sum+node[t<<1|1].sum;
    }
}

int query(int t,int l,int r)
{
    int mid=(node[t].l+node[t].r)>>1,ans=0;
    if(node[t].l==l && node[t].r==r) return node[t].sum;
    if(r<=mid) ans+=query(t<<1,l,r);
    else if(l>mid) ans+=query(t<<1|1,l,r);
    else{
        ans+=query(t<<1,l,mid);
        ans+=query(t<<1|1,mid+1,r);
    }
    return ans;
}

int main()
{
    int t,aaa=1;
    char s[10];
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        memset(node,0,sizeof(node));
        for(int i=1;i<=n;i++) scanf("%d",&num[i]);
        built(1,1,n);
        printf("Case %d:\n",aaa++);
        while(scanf("%s",s) && strcmp(s,"End"))
        {
            scanf("%d %d",&p,&q);
            if(s[0]=='A')
            {
                add(1,1,n);
            }
            else if(s[0]=='S')
            {
                q=-q;
                add(1,1,n);
            }
            else
            {
                int ans=query(1,p,q);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值