HDU 1166 敌兵布阵

循环是人递归是神

于 2020-09-30 13:31:39 发布

阅读量100

点赞数

分类专栏：线段树 HDU

本文链接：https://blog.csdn.net/qq_35752161/article/details/108877335

版权

线段树同时被 2 个专栏收录

4 篇文章 0 订阅

订阅专栏

HDU

4 篇文章 0 订阅

订阅专栏

该博客主要介绍了线段树这种数据结构，并给出了一个C++实现的实例，用于执行区间查询、增加和减少操作。线段树是一种高效的数据结构，能够快速地处理区间上的更新和查询操作。代码中定义了`build`、`update`和`query_range`三个函数，分别用于构建线段树、更新区间值和查询区间和。示例中还包含了主函数`main`，演示了如何使用这些操作处理输入数据。

摘要由CSDN通过智能技术生成

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

常规线段树操作

#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
#include<map>
#include<unordered_map>
#include<set>
#include<unordered_set>
#include<stack>
#include<sstream>
#include<fstream>
using namespace std;

const int maxn = 500000;
int tree[maxn], arr[maxn];
void build(int node, int start, int end) {
	if (start == end) {
		tree[node] = arr[start];
		return;
	}
	int mid = (start + end) / 2;
	build(2 * node + 1, start, mid);
	build(2 * node + 2, mid + 1, end);
	tree[node] = tree[2 * node + 1] + tree[2 * node + 2];
}
 
void update(int node, int start, int end, int idx, int val) {
	if (start == end && start == idx) {
		arr[idx] += val;//Note
		tree[node] += val;//Note
		return;
	}
	int mid = (start + end) / 2;
	int left_node = 2 * node + 1;
	int right_node = 2 * node + 2;
	if (idx >= start && idx <= mid)
		update(left_node, start, mid, idx, val);
	else
		update(right_node, mid + 1, end, idx, val);
	tree[node] = tree[left_node] + tree[right_node];
}


// 区间和查找，lr为查找范围，LR为线段树范围
int query_range(int node, int l, int r, int L, int R) {
	if (l <= L && r >= R)
		return tree[node];
	int mid = (L + R) / 2;
	int sum = 0;
	//当线段树范围[L,mid]和查找范围[l,r]有交集时
	if (mid >= l)
		sum += query_range(node * 2 + 1, l, r, L, mid);
	if (mid < r)
		sum += query_range(node * 2 + 2, l, r, mid + 1, R);
	return sum;
}


int main()
{
	int t;
	cin >> t;
	int cnt = 0;
	while (t--) {
		int n;
		cin >> n;
		for (int i = 0; i < n; i++) 
			cin>>arr[i];
		build(0, 0, n - 1);
		printf("Case %d:\n", ++cnt);
		string s;
		while (cin >> s) {
			if (s == "Query")
			{
				int i, j;
				scanf("%d%d", &i, &j);
				i--; j--;
				printf("%d\n",query_range(0, i, j, 0, n - 1));
			}
			if (s == "Add")
			{
				int i, j;
				scanf("%d%d", &i, &j);
				update(0, 0, n - 1, --i, j);
			}
			if (s == "Sub")
			{
				int i, j;
				scanf("%d%d", &i, &j);
				update(0, 0, n - 1, --i, -j);
			}
			if (s == "End")
				break;
		}
	}
	return 0;
}