社区讨论

90分求调

P3372【模板】线段树 1参与者 4已保存回复 4

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
4 条
当前快照
1 份
快照标识符
@mdo07iz3
此快照首次捕获于
2025/07/29 11:55
7 个月前
此快照最后确认于
2025/07/29 17:36
7 个月前
查看原帖
CPP
#include <bits/stdc++.h>
using namespace std;
int n,q,a[100005];
struct node
{
	int l,r;
	long long sum,tag;
}tree[100005 * 4];
void push_up(int rt)
{
	tree[rt].sum = tree[rt * 2].sum + tree[rt * 2 + 1].sum;
}
void push_down(int rt)
{
	if (!tree[rt].tag)
	{
		return;
	}
	tree[rt * 2].sum += tree[rt].tag * (tree[rt * 2].r - tree[rt * 2].l + 1);
	tree[rt * 2 + 1].sum += tree[rt].tag * (tree[rt * 2 + 1].r - tree[rt * 2  + 1].l + 1);
	tree[rt * 2].tag += tree[rt].tag;
	tree[rt * 2 + 1].tag += tree[rt].tag;
	tree[rt].tag = 0;
}

void build(int rt,int l,int r)
{
	tree[rt].l = l;
	tree[rt].r = r;
	if (l == r)
	{
		tree[rt].sum = a[l];
		return;
	}
	int mid = (l + r) / 2;
	build(rt * 2,l,mid);
	build(rt * 2 + 1,mid + 1,r);
	push_up(rt);
	return;
}
void upd(int rt,int x,int y,int d)
{
	if (x <= tree[rt].l && y >= tree[rt].r)
	{
		tree[rt].sum += (tree[rt].r - tree[rt].l + 1) * d;
		tree[rt].tag += d;
		return;
	}
	push_down(rt);
	int mid = (tree[rt].l + tree[rt].r) / 2;
	if (x <= mid)
	{
		upd(rt * 2,x,y,d);
	}
	if (y > mid)
	{
		upd(rt * 2 + 1,x,y,d);
	}
	push_up(rt);
	return;
}
long long qry_sum(int rt,int x,int y)
{
	if (x <= tree[rt].l && y >= tree[rt].r)
	{
		return tree[rt].sum;
	}
    push_down(rt);
	int mid = (tree[rt].l + tree[rt].r) / 2;
	long long ret = 0;
	if (x <= mid)
	{
		ret += qry_sum(rt * 2,x,y);
	}
	if (y > mid)
	{
		ret += qry_sum(rt * 2 + 1,x,y);
	}
	push_up(rt);
	return ret;
}

int main()
{
	cin >> n >> q;
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
	}
	build(1,1,n);
	int opt,x,y,z;
	for (int i = 1; i <= q; i++)
	{
		cin >> opt;
		if (opt == 1)
		{
			cin >> x >> y >> k;
			upd(1,x,y,k);
		}
		else
		{
			cin >> x >> y;
			cout << qry_sum(1,x,y) << endl;
		}
	}
	return 0;
}

回复

4 条回复,欢迎继续交流。

正在加载回复...