hdu_1166 敌兵布阵（线段树）

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=1166

分析：用数组解这道题，果断超时。传说中的线段树入门题-，-。

线段树基础可以点这里

我的代码（线段树）：

#include<stdio.h>#define  max 50000struct Node{int left;int right;int sum;}segTree[4*max+10];int p[max+10];void CreatsegTree(int index,int lef,int rig) //创建一个[lef,rig]的线段树。{segTree[index].left=lef;segTree[index].right=rig;if(lef==rig){segTree[index].sum=p[lef];return;}int mid=(lef+rig)>>1;int nindex=index<<1;CreatsegTree(nindex,lef,mid);CreatsegTree(nindex+1,mid+1,rig);segTree[index].sum=segTree[nindex].sum+segTree[nindex+1].sum;}void serTree(int index,int k,int num)  //更新[k,k]的位置更新线段树的值。{if(k==segTree[index].left&&k==segTree[index].right) //找到[k,k]为更新。{segTree[index].sum+=num;return;}int mid=(segTree[index].left+segTree[index].right)>>1;if(k<=mid){serTree(index*2,k,num);segTree[index].sum+=num; //底层的位置数更新了，上层的一起更新。}else{serTree(index*2+1,k,num);segTree[index].sum+=num; //底层的位置数更新了，上层的一起更新。}}int ans=0;void Query(int index,int a,int b)  //计算[a,b]段的和。{int mid=(segTree[index].left+segTree[index].right)>>1;if(segTree[index].left==a&&segTree[index].right==b){ans+=segTree[index].sum;return;}if(segTree[index].left==segTree[index].right) return;if(b>mid&&a>mid) Query(index*2+1, a, b);else if(a<=mid&&b<=mid) Query( index*2, a, b);else{Query( index*2, a, mid);Query( index*2+1, mid+1, b);}}int main(){int t;scanf("%d",&t);int cas=1;while(t--){int n,i;scanf("%d",&n);for(i=1;i<=n;i++) scanf("%d",p+i);CreatsegTree(1,1,n);char str[10];printf("Case %d:\n",cas++);while(~scanf("%s",str)&&str[0]!='E'){int a ,b;scanf("%d%d",&a,&b);ans=0;switch(str[0]){case 'A': serTree( 1, a,b);break;case 'S': serTree( 1, a,-b);break;case 'Q':    Query(1, a, b);    printf("%d\n",ans);    break;}}}return 0;}

总结：好久好久以前就遇到这样的题目了，一直没有勇气做，今天终于鼓起勇气做线段树o(≧v≦)o~~

今天来看这个代码··觉得好搓呀，来一个不怎么搓的 o(≧v≦)o~~

#include<stdio.h>#define  MAXN 50000struct Node{int left;int right;int sum;}segTree[4*MAXN+10];int p[MAXN+10];void CreatsegTree(int index,int lef,int rig) //创建一个[lef,rig]的线段树。{segTree[index].left=lef;segTree[index].right=rig;if(lef==rig){segTree[index].sum=p[lef];return;}int mid=(lef+rig)>>1;CreatsegTree(index<<1,lef,mid);CreatsegTree(index<<1|1,mid+1,rig);segTree[index].sum=segTree[index<<1].sum+segTree[index<<1|1].sum;}void ModfiyTree(int index,int k,int num)  //更新[k,k]的位置更新线段树的值。{if(segTree[index].left==k&&k==segTree[index].right) //找到[k,k],更新。{segTree[index].sum+=num;return;}int mid=(segTree[index].left+segTree[index].right)>>1;if(k<=mid){ModfiyTree(index*2,k,num);segTree[index].sum+=num; //底层的位置数更新了，上层的一起更新。}else{ModfiyTree(index*2+1,k,num);segTree[index].sum+=num; //底层的位置数更新了，上层的一起更新。}}//int ans;int   Query(int index,int a,int b)  //计算[a,b]段的和。{    if(segTree[index].left==a && segTree[index].right==b)    {        return segTree[index].sum;    }    //当左右端点相等的时候，即到了叶子结点的位置，要返回了。if(segTree[index].left==segTree[index].right) {return 0;}int mid=(segTree[index].left+segTree[index].right)>>1;if(b<=mid) Query( index<<1, a, b); else if(a>mid) Query(index<<1|1, a, b);else{ return Query( index<<1, a, mid)+Query( index<<1|1, mid+1, b);}}int main(){int t;scanf("%d",&t);int cas=1;while(t--){int n,i;scanf("%d",&n);for(i=1;i<=n;i++) scanf("%d",p+i);CreatsegTree(1,1,n);char str[10];printf("Case %d:\n",cas++);while(~scanf("%s",str)&&str[0]!='E'){int a ,b;scanf("%d%d",&a,&b);switch(str[0]){case 'A': ModfiyTree( 1, a,b);break;case 'S': ModfiyTree( 1, a,-b);break;case 'Q':    printf("%d\n",Query(1, a, b));    break;}}}return 0;}