A Simple Problem with Integers(线段树)

目录

描述

输入

输出

样例输入 

样例输出 

思路

建树 

第一次错误解法(正确解法在下面,可跳过这一步)

正确解法

code 


描述

You have N integers, A1, A2, ... , AN. You need to deal with two kinds of operations. One type of operation is to add some given number to each number in a given interval. The other is to ask for the sum of numbers in a given interval.

输入

The first line contains two numbers N and Q. 1 ≤ N,Q ≤ 100000.
The second line contains N numbers, the initial values of A1, A2, ... , AN. -1000000000 ≤ Ai ≤ 1000000000.
Each of the next Q lines represents an operation.
"C a b c" means adding c to each of AaAa+1, ... , Ab. -10000 ≤ c ≤ 10000.
"Q a b" means querying the sum of AaAa+1, ... , Ab.

输出

You need to answer all Q commands in order. One answer in a line.

样例输入 

10 5
C 3 6 3
Q 2 4

样例输出 

9
15

思路

不要看这题全英文,但这题就是属于线段树模版题,基本做法是一样的

建树 

第一次错误解法(正确解法在下面,可跳过这一步)

树没有更新成功,id=10,id=11理论上是要继续更新他们的sum值,但我的错误代码没有更新

#define _CRT_SECURE_NO_WARNINGS 1
#include<iostream>
using namespace std;
const int N = 100010;
#define ll long long
ll nums[N];
struct Tree {
	int l, r;
	ll sum;
	ll tag;
}tree[N << 2];
void pushup(int u) {
	tree[u].sum = tree[u << 1].sum + tree[u << 1 | 1].sum;
}
void build(int l, int r, int u) {
	if (l == r) tree[u] = { l,r,nums[l] };//要与结构体Tree一一对应 

	else {
		tree[u] = { l,r };
		int mid = (l + r) >> 1;
		build(l, mid, u << 1);
		build(mid + 1, r, u << 1 | 1);
		pushup(u);
	}
}
ll query(int L, int R, int u) {
	int l = tree[u].l, r = tree[u].r;
	if (l >= L && r <= R)return tree[u].sum;

	int mid = (l + r) >> 1;
	ll sum = 0;
	if (L <= mid) sum += query(L, R, u << 1);
	if (R > mid) sum += query(L, R, u << 1 | 1);
	return sum;
}
void pushdown(int u, int ln, int rn) {
	if (tree[u].tag) {
		tree[u << 1].tag += tree[u].tag;
		tree[u << 1 | 1].tag += tree[u].tag;
		tree[u << 1].sum += tree[u].tag * ln;
		tree[u << 1 | 1].sum += tree[u].tag * rn;
		tree[u].tag = 0;
	}
}
void updatespan(int L, int R, int value, int u) {
	int l = tree[u].l, r = tree[u].r;
	int mid = (l + r) >> 1;
	if (l >= L && r <= R) {
		tree[u].tag += value;
		tree[u].sum += value * (r - l + 1);
		
		return;
	}
	pushdown(u, mid - l + 1, r - mid);
	if (L <= mid) updatespan(L, R, value, u << 1);
	if (R > mid) updatespan(L, R, value, u << 1 | 1);
	pushup(u);
}
int main()
{
	int n, q;
	while (scanf("%d%d", &n, &q) != EOF) {
		for (int i = 1; i <= n; i++) scanf("%lld", &nums[i]);
		build(1, n, 1);
		while (q--) {
			string cz;
			int op1, op2;
			cin >> cz;
		
			if (cz == "C") {
				int val;
				scanf("%d%d%d", &op1, &op2, &val);
				updatespan(op1, op2, val, 1);
			}
			else {
				scanf("%d%d", &op1, &op2);
				printf("%lld\n", query(op1, op2, 1));
			}
		}
	}
	return 0;
}

正确解法

关键在于updatespan函数中,if语句中我没有继续pushdown导致错误

  

code 

#define _CRT_SECURE_NO_WARNINGS 1
#include<iostream>
using namespace std;
const int N = 100010;
#define ll long long
ll nums[N];
struct Tree {
	int l, r;
	ll sum;
	ll tag;
}tree[N << 2];
void pushup(int u) {
	tree[u].sum = tree[u << 1].sum + tree[u << 1 | 1].sum;
}
void build(int l, int r, int u) {
	if (l == r) {
		tree[u] = { l,r,nums[l] };//要与结构体Tree一一对应 

	else {
		tree[u] = { l,r };
		int mid = (l + r) >> 1;
		build(l, mid, u << 1);
		build(mid + 1, r, u << 1 | 1);
		pushup(u);
	}
}
ll query(int L, int R, int u) {
	int l = tree[u].l, r = tree[u].r;
	if (l >= L && r <= R)return tree[u].sum;

	int mid = (l + r) >> 1;
	ll sum = 0;
	if (L <= mid) sum += query(L, R, u << 1);
	if (R > mid) sum += query(L, R, u << 1 | 1);
	return sum;
}
void pushdown(int u, int ln, int rn) {
	if (tree[u].tag) {
		tree[u << 1].tag += tree[u].tag;
		tree[u << 1 | 1].tag += tree[u].tag;
		tree[u << 1].sum += tree[u].tag * ln;
		tree[u << 1 | 1].sum += tree[u].tag * rn;
		tree[u].tag = 0;
	}
}
void updatespan(int L, int R, int value, int u) {
	int l = tree[u].l, r = tree[u].r;
	int mid = (l + r) >> 1;
	if (l >= L && r <= R) {
		tree[u].tag += value;
		tree[u].sum += value * (r - l + 1);
		pushdown(u, mid - l + 1, r - mid);//关键
		return;
	}
	pushdown(u, mid - l + 1, r - mid);
	if (L <= mid) updatespan(L, R, value, u << 1);
	if (R > mid) updatespan(L, R, value, u << 1 | 1);
	pushup(u);
}
int main()
{
	int n, q;
	while (scanf("%d%d", &n, &q) != EOF) {
		for (int i = 1; i <= n; i++) scanf("%lld", &nums[i]);
		build(1, n, 1);
		while (q--) {
			string cz;
			int op1, op2;
			cin >> cz;
		
			if (cz == "C") {
				int val;
				scanf("%d%d%d", &op1, &op2, &val);
				updatespan(op1, op2, val, 1);
			}
			else {
				scanf("%d%d", &op1, &op2);
				printf("%lld\n", query(op1, op2, 1));
			}
		}
	}
	return 0;
}

  • 26
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Here is a C++ program that constructs a max heap with integers and prints it in the rotated form: ```cpp #include <iostream> #include <vector> using namespace std; // function to swap two integers void swap(int& a, int& b) { int temp = a; a = b; b = temp; } // function to heapify the given vector void heapify(vector<int>& arr, int n, int i) { int largest = i; // initialize largest as root int left = 2*i + 1; // left child index int right = 2*i + 2; // right child index // if left child is larger than root if (left < n && arr[left] > arr[largest]) largest = left; // if right child is larger than largest so far if (right < n && arr[right] > arr[largest]) largest = right; // if largest is not root if (largest != i) { // swap the root with largest element swap(arr[i], arr[largest]); // recursively heapify the affected sub-tree heapify(arr, n, largest); } } // function to build max heap void buildMaxHeap(vector<int>& arr, int n) { // start from the last non-leaf node and heapify each node for (int i = n / 2 - 1; i >= 0; i--) heapify(arr, n, i); } // function to print the heap in the rotated form void printRotatedHeap(vector<int>& arr, int n) { int height = log2(n) + 1; // height of the heap int index = 0; // current index in the heap int spaces = pow(2, height - 1) - 1; // number of spaces before the first element of the current level // print each level of the heap in the rotated form for (int i = 0; i < height; i++) { // print the spaces before the first element of the current level for (int j = 0; j < spaces; j++) cout << " "; // print the elements of the current level for (int j = 0; j < pow(2, i) && index < n; j++) { cout << arr[index++] << " "; // print the spaces between elements of the current level for (int k = 0; k < 2 * spaces + 1; k++) cout << " "; } // move to the next line and adjust the number of spaces for the next level cout << endl; spaces /= 2; } } int main() { int n; cout << "Enter the number of elements: "; cin >> n; vector<int> arr(n); cout << "Enter the elements: "; for (int i = 0; i < n; i++) cin >> arr[i]; // build max heap buildMaxHeap(arr, n); // print the heap in the rotated form cout << "Max heap in the rotated form:\n"; printRotatedHeap(arr, n); return 0; } ``` In this program, we first define a `swap` function to swap two integers, and a `heapify` function to heapify the sub-tree rooted at a given index `i` in the given vector `arr`. We then define a `buildMaxHeap` function to build the max heap from the given vector `arr`. Finally, we define a `printRotatedHeap` function to print the max heap in the rotated form. In the `main` function, we first read the number of elements and the elements themselves from the user using `cin`. We then build the max heap using `buildMaxHeap` function, and print the heap in the rotated form using `printRotatedHeap` function. The `printRotatedHeap` function uses the height of the heap to determine the number of levels, and the number of spaces before the first element of each level. It then prints each level of the heap in the rotated form, by printing the elements of the level followed by the spaces between elements.

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值