poj 3468 线段树

#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdio>
#include <algorithm>
using namespace std;
const int MAXN = (int)1e5 + 10;
struct Node
{
	long long sum;
	int add;//延迟标记
	int left, right;
	int mid() { return (left + right) >> 1; }
} Tree[MAXN << 2];
int a[MAXN];

void buildTree(int i, int l, int r)
{
	Tree[i].add = 0;
	Tree[i].left = l;
	Tree[i].right = r;
	if (l == r)
	{
		Tree[i].sum = a[l];
		return;
	}
	int m = Tree[i].mid();
	buildTree(i << 1, l, m);
	buildTree(i << 1 | 1, m + 1, r);
	Tree[i].sum = Tree[i << 1].sum + Tree[i << 1 | 1].sum;
}

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

void push_down(int i)
{
	int lchild = i << 1;
	int rchild = i << 1 | 1;
	if (Tree[i].add)
	{
		Tree[lchild].add += Tree[i].add;
		Tree[rchild].add += Tree[i].add;
		Tree[lchild].sum += (long long)Tree[i].add * (Tree[lchild].right - Tree[lchild].left + 1);
		Tree[rchild].sum += (long long)Tree[i].add * (Tree[rchild].right - Tree[rchild].left + 1);
		Tree[i].add = 0;
	}
}

void updata(int i, int c, int l, int r)
{
	if (Tree[i].left == l && Tree[i].right == r)
	{
		Tree[i].add += c;
		Tree[i].sum += (long long)c * (r - l + 1);
		return;
	}
	if (Tree[i].left == Tree[i].right)
		return;
	push_down(i);
	int m = Tree[i].mid();
	if (r <= m)
		updata(i << 1, c, l, r);
	else if (l > m)
		updata(i << 1 | 1, c, l, r);
	else
	{
		updata(i << 1, c, l, m );
		updata(i << 1 | 1, c, m + 1, r);
	}
	push_up(i);
}


long long query(int i, int l, int r)
{
	if (Tree[i].left == l && Tree[i].right == r)
	{
		return Tree[i].sum;
	}
	push_down(i);
	int m = Tree[i].mid();
	if (r <= m)
	{
		return query(i << 1, l, r);
	}
	else if (l > m)
	{
		return query(i << 1 | 1, l, r);
	}
	else
	{
		return query(i << 1, l, m) + query(i << 1 | 1, m + 1, r);
	}
}

int main(void)
{
	ios::sync_with_stdio(false);
	
	int N, Q, x, y, c;
	char op;
	while (cin >> N >> Q)
	{
		for (int i = 1; i <= N; ++i)
			cin >> a[i];
		buildTree(1, 1, N);
		while (Q--)
		{
			cin >> op >> x >> y;
			if (op == 'Q')
				cout << query(1, x, y) << endl;
			else
			{
				cin >> c;
				updata(1, c, x, y);
			}
		}

	}

	return 0;
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值