题目意思:
http://acm.hdu.edu.cn/showproblem.php?pid=1166
线段树的应用,自己的第一个线段树代码,做题之前一直看高级数据结构之县单数并参考了大神代码,自己认为,线段树就是二叉树,只是结点的信息量比较大,主要的是怎么保存结点的信息,这个要根据题目意思进行修改,例如本题要求的数总人数,那么需要保存i-j的总人数,代码如下
AC代码:
/** *线段树的简单应用,用scanf和printf输入输出 *cin和cout会超时 */ #include<iostream> #include<algorithm> #include<cstdio> #include<cstring> using namespace std; int a[50001];//存储结点的位置 struct node{ int left,right;//线段区间 int sum;//区间上的人数 }b[150005]; void Build(int left,int right,int i){//建树 int mid; b[i].left=left; b[i].right=right; if(left==right){//先用数组存下该队的人 b[i].sum=a[left]; return ; } //cout<<"ok"<<endl; mid=(left+right)/2; Build(left,mid,2*i);//递归建立左子树 Build(mid+1,right,2*i+1);//递归建立右子树 b[i].sum=b[2*i].sum+b[2*i+1].sum;//记录该结点左右子树的值 } void Add(int j,int num,int i){ if(b[i].left==b[i].right){//到达叶子结点 b[i].sum+=num; return ;//注意返回 } else{ b[i].sum+=num;//他的所有父节点都增加 if(j<=b[i*2].right) Add(j,num,2*i);//递归左子树 else Add(j,num,2*i+1);//建立右子树 } } int Query(int left,int right,int i){//查询 int mid; if(b[i].left==left&&b[i].right==right) return b[i].sum; mid=(b[i].left+b[i].right)/2; if(right<=mid) return Query(left,right,2*i);//在左子树 else if(left>mid) return Query(left,right,2*i+1);//在右子树 else return Query(left,mid,2*i)+Query(mid+1,right,2*i+1);//否则在中间 } int main() { int t,k=0,i,j,num,n; char s[10]; scanf("%d",&t); while(t--){ scanf("%d",&n); for(i=1;i<=n;i++){ scanf("%d",&a[i]); } //cout<<"Case "<<++k<<":"<<endl; printf("Case %d:\n",++k); Build(1,n,1);//建树 while(1){ scanf("%s",s); if(!strcmp(s,"End")) break; scanf("%d%d",&j,&num); if(!strcmp(s,"Query")){ printf("%d\n",Query(j,num,1)); //cout<<Query(j,num,1)<<endl; } if(!strcmp(s,"Add")) Add(j,num,1);//从根结点开始 if(!strcmp(s,"Sub")) Add(j,-num,1); } } return 0; }
原文地址:http://blog.csdn.net/fool_ran/article/details/41442779