hdoj 1166 敌兵布阵(线段树||树状数组)

敌兵布阵

点击打开链接

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 56829    Accepted Submission(s): 23945

Problem Description

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的责骂的.

 

Input

第一行一个整数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条命令

 

Output

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

 

Sample Input

1101 2 3 4 5 6 7 8 9 10Query 1 3Add 3 6Query 2 7Sub 10 2Add 6 3Query 3 10End 

 

Sample Output

Case 1:63359

 

最为繁琐的线段树写

代码中，我都将他们的操作区间注释了

线段树与树状数组：

两者在复杂度上同级, 但是树状数组的常数明显优于线段树, 其编程复杂度也远小于线段树.树状数组的作用被线段树完全涵盖, 凡是可以使用树状数组解决的问题, 使用线段树一定可以解决, 但是线段树能够解决的问题树状数组未必能够解决.树状数组的突出特点是其编程的极端简洁性, 使用lowbit技术可以在很短的几步操作中完成树状数组的核心操作，与之相关的便是其代码效率远高于线段树。另外，当问题推广到高维情形时高维树状数组有高维线段树所无法企及的常数优势。

#include<cstdio>#include<cstring>#include<cstdlib>#include<iostream>#include<algorithm>using namespace std;#define lson l,mid,2*v//左孩子操作#define rson mid+1,r,2*v+1//右孩子操作const int MAX=50010;int sum[4*MAX];//注意数组范围int num[MAX];void build (int l,int r,int v)//建立一颗线段树（二叉搜索树）{sum[v]=0;if(l==r)//区间:[....l=r...]{  sum[v]=num[l];  return ;}int mid=(l+r)>>1;build(lson);//[l,mid]build(rson);//[mid+1,r]sum[v]=sum[2*v]+sum[2*v+1] ;}void update(int l,int r,int v,int pos,int x)//单个点更新{if(l==r)//[...,l=r,...]{sum[v]+=x;return ;}int mid=(l+r)>>1;if(pos<=mid)//[l,pos,mid,r]update(lson,pos,x);else//[l,mid,pos,r]update(rson,pos,x);sum[v]=sum[2*v]+sum[2*v+1] ;}int query(int l,int r,int v,int lpos,int rpos)//查询{   if(lpos<=l&&r<=rpos)//[lpos,[l,r],rpos]    return sum[v];   int mid=(l+r)>>1;   if(rpos<=mid)//[...[lpos rpos], mid,r];   return query(lson,lpos,rpos);   else if(lpos>mid)//[l,mid,[lpos,rpos]...];   return query(rson,lpos,rpos);   else //if(lpos<=mid&&rpos>=mid)[...,[lpos,mid,rpos],...]   return query(lson,lpos,mid)+query(rson,mid+1,rpos);//[[lpos,mid],...]+[...,[mid+1,rpos]]} int main(){int T,ca=1;scanf("%d",&T);while(T--){int n;scanf("%d",&n);for(int i=1;i<=n;i++)scanf("%d",&num[i]);build(1,n,1);char opr[10];printf("Case %d:\n",ca++);while(scanf("%s",opr)&&opr[0]!='E'){int a,b;scanf("%d%d",&a,&b);if(opr[0]=='Q'){int ans=query(1,n,1,a,b);printf("%d\n",ans);}else         {   if(opr[0]=='S')   b=-b;   update(1,n,1,a,b);    } }}return 0;}

用树状数组写更为简单

这是之前写的代码：

#include<cstdio>#include<cmath>#include<cstring>#include<algorithm>using namespace std;int num[50001];int a[50001];int n;int lowbit(int x)//寻找2^K位{  return x&(-x);//或return (X&(X-1));}void update(int i,int m)//修改i位置上的数（加减某一值m） 加:update(i,m) 减: update(i,-m);{  while(i<=n)  {  num[i]+=m;  i+=lowbit(i);  }}int getsum(int i)//求前n项和类似与查询 {    int sum=0;    while(i>0)//注意不是i>=0    {    sum+=num[i];    i-=lowbit(i);    }   return sum;}int main(){  int T,ca=1;  scanf("%d",&T);  while(T--)  {   scanf("%d",&n);   memset(num,0,sizeof(num));//发现没有这句总是WA  Wa了我6次   for(int i=1;i<=n;i++)   {    int x;scanf("%d",&x);    update(i,x);   }    char s[10];   memset(a,0,sizeof(a));   int j=0;   while(scanf("%s",s)&&strcmp(s,"End")!=0)   {    int l,r;    scanf("%d%d",&l,&r);    if(strcmp(s,"Query")==0)//查询 结果存储在 a[ ]数组中    a[j++]=getsum(r)-getsum(l-1);//此处也可以直接输出getsum(r)-getsum(l-1)的值（但是注意此处题目的格式）     if(strcmp(s,"Add")==0)//加某一值    update(l,r);      if(strcmp(s,"Sub")==0)//减某一值    update(l,-r);   }     printf("Case %d:\n",ca++);    for(i=0;i<j;i++)     printf("%d\n",a[i]);  } return 0;}