线♂段♂树♂性♂感♂代♂码♂在♂线♂求♂调
查看原帖
线♂段♂树♂性♂感♂代♂码♂在♂线♂求♂调
430133
liangbob楼主2023/3/11 22:53
#include <iostream>
#include <cstdio>
#define int long long
using namespace std;
const int N = 1e5 + 10;

struct node
{
	int l, r, d, z;
};
node t[N << 2];
int a[N];

void down(int k)
{
	if(t[k].z)
	{
		int lc = k * 2;
		int rc = k * 2 + 1;
		t[lc].z += t[k].z;
		t[lc].d += t[k].z * (t[lc].r - t[lc].l + 1);
		t[rc].z += t[k].z;
		t[rc].d += t[k].z * (t[rc].r - t[rc].l + 1);
		t[k].z = 0;
	}
}

void build(int k, int l, int r)
{
	t[k].l = l;
	t[k].r = r;
	if(l == r)
	{
		t[k].d = a[l];
		return;
	}
	int mid = (l + r) >> 1;
	int lc = k * 2;
	int rc = k * 2 + 1;
	build(lc, l, mid);
	build(rc, mid + 1, r);
	t[k].d = t[lc].d + t[rc].d;
}

void update(int k, int l, int r, int p)
{
	if(t[k].l == l && t[k].r == r)
	{
		t[k].d += t[k].d * (t[k].r - t[k].l + 1);
		t[k].z += p;
		return;
	}
	down(k);
	int mid = (t[k].l + t[k].r) / 2;
	int lc = k * 2;
	int rc = k * 2 + 1;
	if(r <= mid)
	{
		update(lc, l, r, p);	
	}
	else if(l > mid)
	{
		update(rc, l, r, p);
	}
	else
	{
		update(lc, l, mid, p);
		update(rc, mid + 1, r, p);
	} 
	t[k].d = t[lc].d + t[rc].d;
}

int query(int k, int l, int r)
{
	if(t[k].l == l && t[k].r == r)
	{
		return t[k].d;
	}
	down(k);
	int mid = (t[k].l + t[k].r) / 2;
	int lc = k * 2;
	int rc = k * 2 + 1;
	if(r <= mid)
	{
		return query(lc, l, r);
	}
	else if(l > mid)
	{
		return query(rc, l, r);
	}
	else
	{
		return query(lc, l, mid) + query(rc, mid + 1, r);
	}
}

signed main()
{
	int n;
	cin >> n;
	int q;
	cin >> q;
	for(int i = 1;i <= n;i++) scanf("%lld ", &a[i]);
	build(1, 1, n);
	while(q--)
	{
		int op;
		scanf("%lld", &op);
		if(op == 1)
		{
			int x, y, k;
			scanf("%lld %lld %lld", &x, &y, &k);
			update(1, x, y, k);
		}
		else
		{
			int x, y;
			scanf("%lld %lld", &x, &y);
			printf("%lld\n", query(1, x, y));
		}
	}
	return 0;
}

555连样例都没过。

2023/3/11 22:53
加载中...