P3372 【模板】线段树 1

发布时间:2024年01月18日

题目描述

如题,已知一个数列,你需要进行下面两种操作:

  1. 将某区间每一个数加上 k k k
  2. 求出某区间每一个数的和。

输入格式

第一行包含两个整数 n , m n, m n,m,分别表示该数列数字的个数和操作的总个数。

第二行包含 n n n 个用空格分隔的整数,其中第 i i i 个数字表示数列第 i i i 项的初始值。

接下来 m m m 行每行包含 3 3 3 4 4 4 个整数,表示一个操作,具体如下:

  1. 1 x y k:将区间 [ x , y ] [x, y] [x,y] 内每个数加上 k k k
  2. 2 x y:输出区间 [ x , y ] [x, y] [x,y] 内每个数的和。

输出格式

输出包含若干行整数,即为所有操作 2 的结果。

样例 #1

样例输入 #1

5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4

样例输出 #1

11
8
20

提示

对于 30 % 30\% 30% 的数据: n ≤ 8 n \le 8 n8 m ≤ 10 m \le 10 m10
对于 70 % 70\% 70% 的数据: n ≤ 10 3 n \le {10}^3 n103 m ≤ 10 4 m \le {10}^4 m104
对于 100 % 100\% 100% 的数据: 1 ≤ n , m ≤ 10 5 1 \le n, m \le {10}^5 1n,m105

保证任意时刻数列中所有元素的绝对值之和 ≤ 10 18 \le {10}^{18} 1018

【样例解释】


我的ac代码:

#include<iostream>
#include<bits/stdc++.h>
#define lc p<<1
#define rc p<<1|1
using namespace std;
using ll = long long;
const int N = 1e5 + 10;
int n, m;
int w[N];
struct node {
	int l;
	int r;
	ll sum;
	ll add;
}tr[4*N];
void pushup(int p) {
	tr[p].sum = tr[lc].sum + tr[rc].sum;
}
void pushdown(int p) {
	if (tr[p].add) {
		tr[lc].sum += tr[p].add * (tr[lc].r - tr[lc].l + 1);
		tr[rc].sum += tr[p].add * (tr[rc].r - tr[rc].l + 1);
		tr[lc].add += tr[p].add;
		tr[rc].add += tr[p].add;
		tr[p].add = 0;
	}
}
void build(int p, int l, int r) {
	tr[p] = { l,r,w[l],0 };
	if (l == r)return;
	int mid = l + r >> 1;
	build(lc, l, mid);
	build(rc, mid + 1, r);
	pushup(p);
}
void update(int p, int x, int y, int k) {
	if (x <= tr[p].l && tr[p].r <= y) {
		tr[p].sum += k * (tr[p].r - tr[p].l + 1);
		tr[p].add += k;
		return;
	}
	int mid = tr[p].l + tr[p].r >> 1;
	pushdown(p);
	if (x <= mid)update(lc, x, y, k);
	if (y > mid)update(rc, x, y, k);
	pushup(p);
}
ll query(int p, int x, int y) {
	if (x <= tr[p].l && tr[p].r <= y)return tr[p].sum;
	int mid = tr[p].l + tr[p].r >> 1;
	pushdown(p);
	ll sum = 0;
	if (x <= mid)sum += query(lc, x, y);
	if (y > mid)sum+=query(rc, x, y);
	return sum;
}
int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i++)cin >> w[i];
	build(1, 1, n);
	while (m--) {
		int z, x, y, k;
		cin >> z;
		if (z == 1) {
			cin >> x >> y >> k;
			update(1, x, y, k);
		}
		else {
			cin >> x >> y;
			cout << query(1, x, y) << endl;
		}
	}
}
文章来源:https://blog.csdn.net/qq_60794824/article/details/135679220
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。