[复习]树状数组 敌兵布阵

来源:互联网 发布:淘宝按重量标价 编辑:程序博客网 时间:2024/05/21 21:45

题目背景
HDU 1166

题目描述
C国的死对头A国这段时间正在进行军事演习,所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了 N 个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段,所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动,可能增加或减少若干人手,但这些都逃不C国的监视。

中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek问:“Tidy,马上汇报第 3 个营地到第 10 个营地共有多少?”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动,而Derek每次询问的段都不一样,所以Tidy不得不每次都一个一个营地的去数,很快就精疲力尽了,Derek对Tidy的计算速度越来越不满:”你个死肥仔,算得这么慢,我炒你鱿鱼!”Tidy想:“你自己来算算看,这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下Tidy只好打电话向计算机专家Windbreaker求救,Windbreaker说:“死肥仔,叫你平时做多点Acm题和看多点算法书,现在尝到苦果了吧!”Tidy说:“我知错了。”但Windbreaker已经挂掉电话了。Tidy很苦恼,这么算他真的会崩溃的,聪明的读者,你能写个程序帮他完成这项工作吗?不过如果你的程序效率不够高的话,Tidy还是会受到Derek的责骂的。

输入格式
第一行一个整数 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 条命令。

输出格式
对第 i 组数据,首先输出“Case i:”和回车。
对于每个 Query 询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在 int 以内。

样例数据
输入
1
10
1 2 3 4 5 6 7 8 9 10
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End
输出
Case 1:
6
33
59

分析:树状数组(线段树)模板

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<string>#include<ctime>#include<cmath>#include<algorithm>#include<cctype>#include<iomanip>#include<queue>#include<set>using namespace std;int getint(){    int f=1,sum=0;    char ch;    for(ch=getchar();(ch<'0'||ch>'9')&&ch!='-';ch=getchar());    if(ch=='-')    {        f=-1;        ch=getchar();    }    for(;ch>='0'&&ch<='9';ch=getchar())        sum=(sum<<3)+(sum<<1)+ch-48;    return sum*f;}const int maxn=50010;int T,n,x,y;int a[maxn];char s[10]; int lowbit(int x){    return x & (-x);}void add(int i,int x){    for(int j=i;j<=n;j+=lowbit(j))        a[j]+=x;}void query(int l,int r){    int ans=0;    for(int i=r;i>=1;i-=lowbit(i))        ans+=a[i];    for(int i=l-1;i>=1;i-=lowbit(i))        ans-=a[i];    printf("%d\n",ans);}int main(){    freopen("dbbz.in","r",stdin);    freopen("dbbz.out","w",stdout);    T=getint();    for(int t=1;t<=T;++t)    {        memset(a,0,sizeof(a));        printf("Case %d:\n",t);        n=getint();        for(int i=1;i<=n;++i)        {            x=getint();            add(i,x);        }        scanf("%s",s+1);        while(s[1]!='E')        {            x=getint();y=getint();            if(s[1]=='A')                add(x,y);            if(s[1]=='S')                add(x,-y);            if(s[1]=='Q')                query(x,y);            scanf("%s",s+1);        }    }    return 0;} 

本题结。

原创粉丝点击