hdu1166 敌兵布阵 树状数组

第一行一个整数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 表示结束,这条命令在每组数据最后出现;


还是查询和更新操作,树状数组~

#include<stdio.h>
#include<string.h>
#include<math.h>
int c[50005];
int n;
int lowbit(int x)
{
    return x&(-x);
}
void modify(int i,int j)//修改
{
    while(i<=n)
    {
        c[i]+=j;
        i=i+lowbit(i);
    }
}
long long sum(int i)//求和
{
    long long s=0;
    while(i>0)
    {
        s+=c[i];
        i=i-lowbit(i);
    }
    return s;
}
int main()
{
    int T,a;
    scanf("%d",&T);
    for(int t=1;t<=T;t++)
    {
        memset(c,0,sizeof(c));
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a);
            modify(i,a);
        }
        char str[10];
        scanf("%s",str);
        printf("Case %d:\n",t);
        while(strcmp(str,"End")!=0)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            if(strcmp(str,"Query")==0)
                printf("%lld\n",sum(y)-sum(x-1));
            else if(strcmp(str,"Add")==0)
                modify(x,y);
            else if(strcmp(str,"Sub")==0)
                modify(x,(-1)*y);
            scanf("%s",str);
        }
    }
}




  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值