hdu_1166 敌兵布阵（线段树）

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

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

线段树基础可以点这里

我的代码（线段树）：

#include<stdio.h>
#define  max 50000
struct 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 50000
struct 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;
}