【数据结构】 线段树 区间修改&区间查询 cdoj1057

题目链接:https://acm.uestc.edu.cn/problem

Description 

 

秋实大哥是一个儒雅之人,昼听笙歌夜醉眠,若非月下即花前。

所以秋实大哥精心照料了很多花朵。现在所有的花朵排成了一行,每朵花有一个愉悦值。

秋实大哥每天要对着某一段连续的花朵歌唱,然后这些花朵的愉悦值都会增加一个相同的值vvv(vvv可能为负)。

同时他想知道每次他唱完歌后这一段连续的花朵的愉悦值总和是多少。

Standard Input

第一行有一个整数nnn,表示花朵的总数目。

第二行包含nnn个整数a_ia\_ia_i,表示第iii朵花初始的愉悦值。

第三行包含一个整数mmm,表示秋实大哥唱了mmm天的歌。

接下来mmm行,每行包含三个整数lll rrr vvv,表示秋实大哥对着[l,r][l, r][l,r]这个区间内的花朵歌唱,每朵花的愉悦值增加了vvv。

1≤n,m,a_i,∣v∣≤1000001\leq n,m,a\_i,|v|\leq 1000001≤n,m,a_i,∣v∣≤100000,1≤l≤r≤n。1\leq l\leq r\leq n。1≤l≤r≤n。

Standard Output

输出共mmm行,第iii行表示秋实大哥完成第iii天的歌唱后,那一段花朵的愉悦值总和。

Samples

InputOutput
3
0 0 0
3
1 2 1
1 2 -1
1 3 1

 区间查询&区间修改

学习了一波lazy思想:以区间修改为例,当前节点标记一个lazy值(要改的值),如果当前区间完全包含在修改区间内,那么这个点的sum加上(区间长度*lazy值),后面的值就不用再一一修改了,当不能完全包含的时候,就把lazy值往下传一个一个改。

代码如下:

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5;
int a[maxn], n, q;
struct node {
	int l, r;
	long long sum, lazy;
	void update(long long  x)
	{
		sum += 1ll*(r - l + 1)*x;
		lazy += x;
	}
}tree[4*maxn];

void push_up(int x)
{
	tree[x].sum = tree[x << 1].sum + tree[x << 1 | 1].sum;
}

void push_down(int x)
{
	int lz = tree[x].lazy;
	if (lz)
	{
		tree[x << 1].update(lz);
		tree[x << 1 | 1].update(lz);
		tree[x].lazy = 0;
	}
}

void build(int x, int l, int r) {
	tree[x].l = l, tree[x].r = r;
	tree[x].sum = tree[x].lazy = 0;
	if (l == r) {
		tree[x].sum = a[l];
	}
	else {
		int mid = (l + r) / 2;
		build(x << 1, l, mid);
		build(x << 1 | 1, mid + 1, r);
		push_up(x);
	}
}

void update(int x, int l, int r, long long val)
{
	int L = tree[x].l, R = tree[x].r;
	if (l <= L && R <= r)
	{
		tree[x].update(val);
	}
	else {
		push_down(x);
		int mid = (L + R) / 2;//当前节点区间的中点值
		if (l <= mid) update(x << 1, l, r,val);
		if (mid < r)update(x << 1 | 1, l, r, val);
		push_up(x);
	}
}

long long query(int x, int l, int r)
{

	int L = tree[x].l, R = tree[x].r;
	if (l <= L && R <= r)
	{
		return tree[x].sum;
	}
	else {
		push_down(x);
		long long ans = 0;
		int mid = (L+ R) / 2;
		if (l <= mid) ans+=query(x << 1, l, r);
		if (mid < r)ans+=query(x << 1 | 1, l, r);
		push_up(x);
		return ans;
	}
}
int main()
{
	scanf("%d", &n);
	for (int i = 1;i <= n;i++)
		scanf("%d", &a[i]);
	build(1, 1, n);
	scanf("%d", &q);
	while (q--)
	{
		int l, r, v;
		scanf("%d%d%d", &l, &r, &v);
		update(1, l, r, v);
		printf("%lld\n", query(1, l, r));
	}
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值