hdu 1166 树状数组

#include <iostream>
#include <stdio.h>
#include <string>
using namespace std;
int lowbit(int x)					//make sure the set of the array  求最小幂2^k
{
	return x&(-x);
}
int sum(int end ,int all[])			//solve sum 求前n 项和
{
	int sum=0;
	while(end > 0)
	{
		sum+=all[end];
		end-=lowbit(end);
	}
	return sum;
}
void puls(int set, int num, int all[], int n)			//array number increase or decrease 某个元素进行加法操作
{
	while(set <= n)										//网上这里有等于，这个是对的，所以一开始数组需要开大一点
	{
		all[set] += num;
		set += lowbit(set);
	}
}
struct data	
{
	string str;
	int i,j;
};
int main()
{
	int T, count=0;
	scanf("%d",&T);
	while(T--)
	{
		int all[50500];
		int N;
		memset(all, 0, sizeof(all) );
		scanf("%d",&N);
		for(int i=1; i<=N; i++)						
			/*
			因为题目的关系这里从1开始输入数据，以便在计算总和时把漏掉的i加上时不会越界,并且必须从1开始“是因为跟那个“将k 化为二进制数时末尾0 的个 数”有关”
			*/
		{
			int temp;
			scanf("%d",&temp);
			puls(i ,temp, all, N);
		}
		
		data temp;
		printf("Case %d:
",++count);
		while(cin>>temp.str && temp.str!="End")
		{
			scanf("%d%d",&temp.i,&temp.j);
			if(temp.str =="Query" )
			{
				
				printf("%d
",sum(temp.j,all)-sum(temp.i-1,all) );		
				/*
				i-1 pay attention 
				the problem describe solve j-i ( Query i j ,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数;)
				so all[i] could be ignore !
				*/
			}
			if(temp.str == "Add" )
			{
				puls(temp.i ,temp.j, all, N);
			}
			if(temp.str =="Sub")
				puls(temp.i ,-temp.j, all, N);
		}
		
	}
	return 0;
}
/*
	第一行一个整数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
*/	

来自为知笔记(Wiz)

附件列表