关于树状数组区间修改区间查询



v#include<iostream>
#include<cstdio>
using namespace std;
long long int c[100005][2] = { 0 };
int n, q;
int lowbit(int x)
{
	return x&(-x);
}
void add(int pos, int x,int f)
{
	while (pos <= n)
	{
		c[pos][f] += x;
		pos += lowbit(pos);
	}
}
long long qu(int pos, int f)
{
	long long res = 0;
	while (pos > 0)
	{
		res += c[pos][f];
		pos -= lowbit(pos);
	}
	return res;
}
long long ask(int pos)
{
	long long res = (pos + 1)*qu(pos, 0) - qu(pos, 1);
	return res;
}
int main()
{
	ios::sync_with_stdio(false);
	cin >> n;
	int a = 0, b, opt, x;
	for (int s = 1; s <= n; s++)
	{
		cin >> b;
		add(s, b - a, 0);
		add(s, (b - a)*s, 1);
		a = b;
	}
	cin >> q;
	for (int s = 1; s <= q; s++)
	{
		cin >> opt;
		if (opt == 1)
		{
			cin >> a >> b >> x;
			add(a, x, 0);
			add(b + 1, -x, 0);
			add(a, x*a, 1);
			add(b + 1, -x*(b + 1), 1);
		}
		else if (opt == 2)
		{
			cin >> a >> b;
			cout << ask(b) - ask(a - 1) << endl;
		}
	}
	return 0;
}



全部评论

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务