HDU-1166(线段树入门)

这个是很简单的一道线段树题目,当然肯定没有树状数组来这爽,但是作为练手的题目还是不错的...

分享给大家我的代码..大家看看有什么可取之处的没有

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>

int N;
struct Node{
	int a;
	int b;
	int val;
}tree[222222];

void Pushup(int i)
{
	tree[i].val=tree[i<<1].val+tree[i<<1|1].val;
}

void Buildtree(int i,int a,int b)
{
	tree[i].a=a;
	tree[i].b=b;
	if(a==b)
	{
		scanf("%d",&tree[i].val);
		return;
	}
	int mid=(a+b)>>1;
	Buildtree(i<<1,a,mid);
	Buildtree(i<<1|1,mid+1,b);
	Pushup(i);
}
void add(int i,int a,int b)
{
	if(tree[i].a==a&&tree[i].b==a)
	{
		tree[i].val+=b;
		return;
	}
	int mid=(tree[i].a+tree[i].b)>>1;
	if(a<=mid)
		add(i<<1,a,b);
	else
		add(i<<1|1,a,b);
	Pushup(i);
}

int query(int i,int a,int b)
{
	if(a<=tree[i].a&&b>=tree[i].b)
	{
		return tree[i].val;
	}
	int mid=(tree[i].a+tree[i].b)>>1;
	int ans;
	if(b<=mid)
		ans=query(i<<1,a,b);
	else if(a>mid)
		ans=query(i<<1|1,a,b);
	else
		ans=query(i<<1,a,mid)+query(i<<1|1,mid+1,b);
	return ans;
}

int main()
{
	int T;
	scanf("%d",&T);
	for(int k=1;k<=T;k++)
	{
		scanf("%d",&N);
		Buildtree(1,1,N);
		char s[10];
		printf("Case %d:\n",k);
		while(scanf("%s",s),strcmp(s,"End"))
		{
			int a,b;
			scanf("%d%d",&a,&b);
			if(strcmp(s,"Query")==0)
			{
				int ans=query(1,a,b);
				printf("%d\n",ans);
			}
			else if(strcmp(s,"Sub")==0)
			{
				add(1,a,-b);
			}
			else if(strcmp(s,"Add")==0)
			{
				add(1,a,b);
			}
		}
	}
	return 0;
}