hdu 1166

hdu 1166 敌兵布阵

 

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

题目大意：

 A国的工兵营地正在进行军事演习。C国是A国的死对头，他们派来了间谍Tidy。Tidy要随时向上级汇报某一段连续的工兵营地内一共有多少人，而且每一个工兵营地的人数都可以变动。苦恼的他还是找到了你。

 兵营的人数可以增加，可以减少。

 

思路：

    区间求和问题

 问题可以转化成求一段连续子区间的和，如果直接线性求和的话，复杂度会在O(N)。虽然使用数组改变兵营人数的复杂度是O(1),但是如果兵营的数目太多的话，程序的效率还是很低效的。如果使用线段树来求解，用树上的每一个节点存储它所对应区间的和，那么求和的复杂度会降至O(logN)，这是一个非常大的飞跃！虽然改变兵营人数操作的复杂度是O(logN)，不如O(1)，但我们程序的瓶颈在求和上！

 

算法步骤：

 

①：初始化线段树，树上的每一个节点值为0.

 输入数列，并不断更新树上节点的值

②：

 Query :  返回i~j的连续区间和

 Add   :  将第i个兵营的人数加上j

 Sub   :  将第i个兵营的人数减少j

 

算法复杂度：

 

①：N个最后一层节点初始化，每个节点更新的复杂度为O(logN).   --- O(NlogN)

②：Query、Add、Sub 操作的复杂度均为O(logN)

则总的复杂度为 O(NlogN)

codes:

/**********************/
/*      hdu 1166       */
/*线段树求解区间和问题**/
/**********************/
 
#include <cstdio>
#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
 
int N,n;
string str;
const int maxn = 1<<17;
int dat[2*maxn-1];
 
void init() {
	n = 1;
	while(n < N)
		n *= 2;
	for(int i = 0 ; i < 2*n - 1 ; i ++)
	{
		dat[i] = 0;
	}
}
 
void update(int k , int ele) {
	k += n-1;
	dat[k] += ele;
	while(k>0)
	{
		k = (k-1)/2;
		dat[k] = dat[2*k+1] + dat[2*k+2];
	}
}
 
int query(int a, int b, int l , int r , int k) {
	if(b <= l || a >= r)
		return 0;
	else if(a <= l && b >= r)
		return dat[k];
	else
	{
		int left = query(a,b,l,(l+r)/2,2*k+1);
		int right = query(a,b,(l+r)/2,r,2*k+2);
		return left + right;
	}
}
 
int main() {
	int CASE;
	int cnt = 1;
	scanf("%d",&CASE);
	while(CASE--)
	{
		printf("Case %d:\n",cnt);
		scanf("%d",&N);
		
		init();
		
		int tmp;
		for(int i = 0 ; i < N ; i ++)
		{
			scanf("%d",&tmp);
			update(i,tmp);
		}
		int a,b;
		while(cin >> str && str != "End")
		{
			if(str == "Query")
			{
				scanf("%d%d",&a,&b);
				printf("%d\n",query(a-1,b,0,n,0));
			}
			else if(str == "Add")
			{
				scanf("%d%d",&a,&b);
				update(a-1,b);
			}
			else if(str == "Sub")
			{
				scanf("%d%d",&a,&b);
				update(a-1,-1*b);
			}
		}
		cnt ++;
	}
	return 0;
}

运行结果：Accept

Time  : 250 MS  

Memory: 808 K