poj 3468 线段树延迟标记

思路:使用线段树时,对于有时不需要立即更新的 子区间,可以先将要更新的内容放在 结点的mark标记上,只要
使用到这个 子区间的时候,再将子区间的sum值进行更新,这样就可以省去大量时间

注意 结点的mark值不为零,说明该区间下的子区间sum值没有更新!


#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <string.h>
#include <algorithm>

using namespace std;
typedef __int64 LL;

const __int64 maxn = 100000 + 10;

struct node
{
	LL mark, sum;			//懒惰标记为mark ,当mark值为零时,说明改子区间的sum值已经更新过
}tree[maxn * 3];

__int64 A[maxn];			//存储输入序列
int n, m;				//数字个数n,询问数m


void build_tree(__int64 lp,__int64 rp,__int64 i)//构建以i为根,区间为[l,,,r]的线段树
{
	if (lp == rp)
	{
		tree[i].sum = A[lp];
		return;
	}

	__int64 mid = (lp + rp) / 2;
	build_tree(lp, mid, i * 2);
	build_tree(mid + 1, rp, i * 2 + 1);
	tree[i].sum = tree[2 * i].sum + tree[2 * i + 1].sum;
}


void update(__int64 lp, __int64 rp, __int64 i)			//用标记法维护线段树,根i对应区间为[lp....rp],目的是更新[lp,rp]下子区间的 sum
{
	if (tree[i].mark == 0)			//如果树根i mark==0 说明该节点下子区间已经 更新过了
		return;

	//否则 要更新该区间下的左右子区间,之后将该节点的mark标记下传给左右子区间
	__int64 mid = (lp + rp) / 2;
	tree[i * 2].sum += tree[i].mark * (mid - lp + 1);
	tree[i * 2 + 1].sum += tree[i].mark * (rp - mid);

	tree[i * 2].mark += tree[i].mark;
	tree[i * 2 + 1].mark += tree[i].mark;
	tree[i].mark = 0;
}

LL query(__int64 tl, __int64 tr, __int64 lp, __int64 rp, __int64 i)			//计算线段树i 对应区间[lp,,rp] 内子区间[tl,tr]上的数字和
{
	if (tl > rp || tr < lp)				//区间[tl,tr]与[lp,rp]无交集
		return 0;
	if (tl <= lp && rp <= tr)			//区间[tl,tr]完全覆盖 [lp,rp]
		return tree[i].sum;

	update(lp, rp, i);				//更新下 区间[lp,rp]下 子区间的sum
	__int64 mid = (lp + rp) / 2;

	return query(tl, tr, lp, mid, i * 2) + query(tl, tr, mid + 1, rp, 2 * i + 1);
}

void add_value(__int64 tl, __int64 tr, __int64 lp, __int64 rp, __int64 i, __int64 val)		//线段树i[lp,rp] 中区间[tl,tr] 的每个数 + val 
{
	if (tl > rp || tr < lp)
		return;
	if (tl <= lp && rp <= tr) //若区间[tl,rp] 覆盖[lp,rp] 
	{
		tree[i].sum += val * (rp - lp + 1);
		tree[i].mark += val;
		return;
	}

	//否则,需要自底向上 对tree[i]进行更新
	update(lp, rp, i);
	__int64 mid = (lp + rp) / 2;
	add_value(tl, tr, lp, mid, i * 2, val);
	add_value(tl, tr, mid + 1, rp, i * 2 + 1, val);

	tree[i].sum = tree[i * 2].sum + tree[i * 2 + 1].sum;
}


void work()
{
	memset(tree,0, sizeof(tree));
	build_tree(1, n, 1);

	char ch;
	__int64 lp, rp, c;
	for (__int64 i = 0; i < m; i++)
	{
		getchar();
		scanf("%c", &ch);
		if ('Q' == ch)
		{
			scanf("%I64d %I64d", &lp, &rp);
			__int64 ans = query(lp, rp, 1, n, 1);
			//cout << ans << endl;
			printf("%I64d\n", ans);
		}
		else
		{
			//cin >> lp >> rp >> c;
			scanf("%I64d %I64d %I64d", &lp, &rp, &c);
			add_value(lp, rp, 1, n, 1, c);
		}
	}
}


int main()
{
	scanf("%d%d", &n, &m);

	//for (__int64 i = 1; i <= n; i++)
	//	cin >> A[i];
	for (__int64 i = 1; i <= n; i++)
		scanf("%I64d", &A[i]);
	//getchar();
	work();
	return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值