HDU 1166 敌兵布阵 一维树状数组or线段树

欢迎参加——每周六晚的BestCoder（有米！）

敌兵布阵 Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others) Total Submission(s): 60962    Accepted Submission(s): 25817 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 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   Sample Output Case 1: 6 33 59   Author Windbreaker   Recommend Eddy   |   We have carefully selected several similar problems for you:   1394  1698  1754  1542  1540  树状数组(Binary Indexed Tree(BIT), Fenwick Tree)是一个查询和修改复杂度 都为log(n)的数据结构。 主要用于查询任意两位之间的所有元素之和，但是每次只能修改一个元素的值；经过简单修改可以在log(n) 的复杂度 下进行范围修改，但是这时只能查询其中一个元素的值。 这种数据结构（算法）并没有C++和Java的库支持，需要自己手动实现。在Competitive Programming的 竞赛中被广 泛的使用。树状数组和线段树很像，但能用树状数组解决的问题，基本上都能用线段树解决， 而线段树能解决的树 状数组不一定能解决。相比较而言，树状数组效率要高很多。                -----------百度百科 经典图： 树状数组的好处在于能快速求任意区间的和a[i]+a[i+1]+……+a[j] 可以设sum（k）=a[1]+a[2]+……+a[k]  则a[x]+a[x+1]+……+a[y]=sum(y)-sum（i-1) 有了树状数组就可以在O（logN）的时间内部求出sum（K） 更新一个a的元素的时间也是O（logN）why？因为数组a跟新了树状数组c也要跟新 sum（k）=c[n1]+c[n2]+c[n3]+……+c[nm] nm=kn(i-1)=ni-lowbit(ni)&&ni-lowbit(ni)<=0 如 sum(6)=c[4]+c[6] lowbit(x)实际上就是x的二进制表示留下最右边的一其他位都变成0 设节点编号为x，那么这个节点管辖的区间为2^k（其中k为x二进制末尾0的个数）个元素。 因为这个区间最后一个   元素必然为Ax，所以很明显：Cn = A(n – 2^k + 1) + ... + An 树状数组适合单个元素经常修改且还要反复要求区间和的情况。  ACcode： #include <map> #include <queue> #include <cmath> #include <cstdio> #include <cstring> #include <stdlib.h> #include <iostream> #include <algorithm> #define maxn 50000+5 using namespace std; int c[maxn]; int n,t,cnt,d,z=1; int lowbit(int k){ return k&(-k); } void get_num(int x,int y){ while(x<=n){ c[x]+=y; x+=lowbit(x); } } int sum(int x){ int s=0; while(x>0){ s+=c[x]; x-=lowbit(x); } return s; } int main(){ scanf("%d",&t); while(t--){ scanf("%d",&n); memset(c,0,sizeof(c)); for(int i=1;i<=n;++i){ scanf("%d",&d); get_num(i,d); } char a[10]; int x,y; printf("Case %d:\n", z++); while(scanf("%s",a)){ if(a[0]=='E') break; scanf("%d%d",&x,&y); if(a[0]=='Q') printf("%d\n",sum(y)-sum(x-1)); else if(a[0]=='S') get_num(x,-y); else get_num(x,y); } } return 0; }   其实这道题也可以用线段树来做 线段树是一种二叉搜索树，与区间树相似，它将一个区间划分成一些单元区间，每个单元区间对应线段树中的一个叶结点。 使用线段树可以快速的查找某一个节点在若干条线段中出现的次数，时间复杂度为O(logN）。而未优化的空间复杂度为2N， 因此有时需要离散化让空间压缩。 如图; ACcode: #pragma warning(disable:4786)//使命名长度不受限制 #pragma comment(linker, "/STACK:102400000,102400000")//手工开栈 #include <map> #include <set> #include <queue> #include <cmath> #include <stack> #include <cctype> #include <cstdio> #include <cstring> #include <stdlib.h> #include <iostream> #include <algorithm> #define rd(x) scanf("%d",&x) #define rd2(x,y) scanf("%d%d",&x,&y) #define rds(x) scanf("%s",x) #define rdc(x) scanf("%c",&x) #define ll long long int #define maxn 50050 #define mod 1000000007 #define INF 0x3f3f3f3f //int 最大值 #define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;++i) #define MT(x,i) memset(x,i,sizeof(x)) #define PI acos(-1.0) #define E exp(1) using namespace std; int tree[maxn<<2]; int a[maxn],loop,n,cnt=1,x,y; char q[20]; void creates(int node,int left,int right){ if(left==right){ tree[node]=a[left]; return; } int m=(left+right)>>1; int temp=node<<1; creates(temp,left,m); creates(temp+1,m+1,right); tree[node]=tree[temp]+tree[temp+1]; } void updata(int node,int p,int add,int left,int right){ if(left==right){ tree[node]+=add; return; } int m=(left+right)>>1; int temp=node<<1; if(p<=m) updata(temp,p,add,left,m); else updata(temp+1,p,add,m+1,right); tree[node]=tree[temp]+tree[temp+1]; } int query(int node,int left,int right,int L,int R){ if(left>=L&&right<=R)return tree[node]; int m=(left+right)>>1; int temp=node<<1; int ret=0; if(L<=m) ret+=query(temp,left,m,L,R); if(R>m) ret+=query(temp+1,m+1,right,L,R); return ret; } int main(){ rd(loop); while(loop--){ rd(n);FOR(i,1,n)rd(a[i]); creates(1,1,n); printf("Case %d:\n",cnt++); while(rds(q)&&q[0]!='E'){ rd2(x,y); if(q[0]=='Q') printf("%d\n",query(1,1,n,x,y)); else { if(q[0]=='A')updata(1,x,y,1,n); else updata(1,x,-y,1,n); } } } return 0; }