HDU 1166 敌兵布阵

Problem Description

C国的死对头A国这段时间正在进行军事演习,所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段,所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动,可能增加或减少若干人手,但这些都逃不过C国的监视。
中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek问:“Tidy,马上汇报第3个营地到第10个营地共有多少人!”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动,而Derek每次询问的段都不一样,所以Tidy不得不每次都一个一个营地的去数,很快就精疲力尽了,Derek对Tidy的计算速度越来越不满:"你个死肥仔,算得这么慢,我炒你鱿鱼!”Tidy想:“你自己来算算看,这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下,Tidy只好打电话向计算机专家Windbreaker求救,Windbreaker说:“死肥仔,叫你平时做多点acm题和看多点算法书,现在尝到苦果了吧!”Tidy说:"我知错了。。。"但Windbreaker已经挂掉电话了。Tidy很苦恼,这么算他真的会崩溃的,聪明的读者,你能写个程序帮他完成这项工作吗?不过如果你的程序效率不够高的话,Tidy还是会受到Derek的责骂的.

Input

第一行一个整数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

树状数组直接应用即可。

#include<iostream>  
#include<algorithm>
#include<math.h>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
const int maxn = 100005;
const int low(int x){ return (x&-x); }
int a[maxn], n, t, m, i, j;
string s;

void add(int x,int y)
{
	for (int i = x; i <= n; i += low(i)) a[i] += y;
}

long long sum(int x)
{
	long long total = 0;
	for (int i = x; i > 0; i -= low(i)) total += a[i];
	return total;
}

int main(){
	while (~scanf("%d", &t))
	{
		int p = 0;
		while (++p <= t)
		{
			cout << "Case " << p << ":" << endl;
			cin >> n;
			memset(a, 0, sizeof(a));
			for (int i = 1; i <= n; i++){ scanf("%d", &m); add(i, m); }
			while (1)
			{
				cin >> s;
				if (s == "End") break;
				else
				{
					scanf("%d%d", &i, &j);
					if (s == "Add") add(i, j);
					if (s == "Sub") add(i, -j);
					if (s == "Query") cout << sum(j) - sum(i - 1) << endl;
				}
			}
		}
	}
	return 0;
}

zkw线段树
#include<iostream>  
#include<algorithm>
#include<math.h>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
const int maxn = 100005;
int a[2 * maxn], n, t, m, i, j, M;
string s;

void add(int x, int y)
{
	x = x + M;	a[x] += y;
	for (x >>= 1; x; x >>= 1) a[x] = a[x + x] + a[x + x + 1];
}

int find(int l, int r)
{
	int tot = 0;
	for (l += M - 1, r += M + 1; l ^ r ^ 1; l >>= 1, r >>= 1)
	{
		if (~l & 1) tot += a[l ^ 1];
		if (r & 1) tot += a[r ^ 1];
	}
	return tot;
}

int main(){
	while (~scanf("%d", &t))
	{
		int p = 0;
		while (++p <= t)
		{
			cout << "Case " << p << ":" << endl;
			cin >> n;
			for (M = 1; M < n; M += M);
			memset(a, 0, sizeof(a));
			for (int i = 1; i <= n; i++){ scanf("%d", &m); add(i, m); }
			while (1)
			{
				cin >> s;
				if (s == "End") break;
				else
				{
					scanf("%d%d", &i, &j);
					if (s == "Add") add(i, j);
					if (s == "Sub") add(i, -j);
					if (s == "Query") cout << find(i, j) << endl;
				}
			}
		}
	}
	return 0;
}

学了splay过来练练手

#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<iostream>
#include<algorithm>
#include<bitset>
#include<functional>
using namespace std;
typedef unsigned long long ull;
typedef long long LL;

struct SPLAY
{
	const static int maxn = 1e5 + 10;
	const static int INF = 0x7FFFFFFF;
	int ch[maxn][2], F[maxn], U[maxn], C[maxn], A[maxn], sz;
	int Node(int f, int u, int c) { A[sz] = C[sz] = c; ch[sz][0] = ch[sz][1] = 0; F[sz] = f; U[sz] = u; return sz++; }
	void clear(){ sz = 1; ch[0][0] = ch[0][1] = C[0] = A[0] = U[0] = F[0] = 0; }
	void rotate(int x, int k)
	{
		int y = F[x];	ch[y][!k] = ch[x][k]; F[ch[x][k]] = y; 
		if (F[y]) ch[F[y]][y == ch[F[y]][1]] = x;
		F[x] = F[y];    F[y] = x;	ch[x][k] = y;		
		C[x] = C[y];	C[y] = C[ch[y][0]] + C[ch[y][1]] + A[y];
	}
	void Splay(int x, int r)
	{
		for (int fa = F[r]; F[x] != fa;)
		{
			if (F[F[x]] == fa) { rotate(x, x == ch[F[x]][0]); return; }
			int y = x == ch[F[x]][0], z = F[x] == ch[F[F[x]]][0];
			y^z ? (rotate(x, y), rotate(x, z)) : (rotate(F[x], z), rotate(x, y));
		}
	}
	bool insert(int &x, int u, int v)
	{
		if (!x) { x = Node(0, u, v); return false; }
		int now = 0;
		for (int i = x; i&&!now; C[i] += v, i = ch[i][U[i] < u])
		{
			if (u == U[i]) { A[i] += v; Splay(i, x); x = i; return true; }
			if (!ch[i][U[i] < u]) now = ch[i][U[i] < u] = Node(i, u, v);
		}
		Splay(now, x);  x = now; return false;
	}
	void find(int &x, int l, int r)
	{
		for (int i = x; i; i = ch[i][U[i] < l - 1])
		{
			if (U[i] == l - 1){ Splay(i, x); x = i; break; }
		}
		for (int i = ch[x][1]; i; i = ch[i][U[i] < r + 1])
		{
			if (U[i] == r + 1){ Splay(i, ch[x][1]); break; }
		}
		printf("%d\n", C[ch[ch[x][1]][0]]);
	}
}tree;

int main()
{
	int n, root, x, T, l, r, t = 0;
	char s[10];
	cin >> T;
	while (T--)
	{
		scanf("%d", &n);
		tree.clear();	root = 0;
		for (int i = 1; i <= n; i++) scanf("%d", &x), tree.insert(root, i, x);
		tree.insert(root, 0, 0);	tree.insert(root, n + 1, 0);
		printf("Case %d:\n", ++t);
		while (scanf("%s", s), s[0] != 'E')
		{
			scanf("%d%d", &l, &r);
			if (s[0] == 'A') tree.insert(root, l, r);
			if (s[0] == 'S') tree.insert(root, l, -r);
			if (s[0] == 'Q') tree.find(root, l, r);
		}
	}
	return 0;
}


你可能感兴趣的:(线段树,HDU,树状数组)