B - 这题就是线段树

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define endl '\n'
#define ll long long
#define int long long
using namespace std;
const int N = 1e5 + 10, mod = 1e9 + 7;
struct Tree
{
	int l, r;
	int sum, xsum, lazy;
} tr[N << 2];
int w[N];
int n, m;
void change(int u, int lazy)
{
	tr[u].xsum = (tr[u].xsum%mod + 2 * tr[u].sum * lazy % mod + (tr[u].r - tr[u].l + 1) * lazy % mod * lazy % mod) % mod;
	tr[u].sum = (tr[u].sum%mod + (tr[u].r - tr[u].l + 1) * lazy % mod) % mod;
	tr[u].lazy += lazy;
	tr[u].lazy %= mod;
}
// void change(int u, int x)
// {
//     tr[u].xsum = tr[u].xsum + 2 * tr[u].sum * x % mod + (tr[u].r - tr[u].l + 1) * x % mod * x % mod;
//     tr[u].xsum %= mod;
//     tr[u].sum = tr[u].sum + (tr[u].r - tr[u].l + 1) * x;
//     tr[u].sum %= mod;
//     tr[u].lazy += x;
//     tr[u].lazy %= mod;
// }
void pushdown(int u)
{
	if (tr[u].lazy)
	{
		change(u << 1, tr[u].lazy);
		change(u << 1 | 1, tr[u].lazy);
		tr[u].lazy = 0;
	}
}
void pushup(int u)
{
	tr[u].sum = (tr[u << 1].sum % mod + tr[u << 1 | 1].sum % mod) % mod;
	tr[u].xsum = (tr[u << 1].xsum % mod + tr[u << 1 | 1].xsum % mod) % mod;
}
void build(int u, int l, int r)
{
	if (l == r)
		tr[u] = {l, r, w[l], (int)((ll)w[l] * w[l] % mod), 0};
	else
	{
		tr[u] = {l, r};
		int mid = l + r >> 1;
		build(u << 1, l, mid);
		build(u << 1 | 1, mid + 1, r);
		pushup(u);
	}
}
void modify(int u, int l, int r, int x)
{
	if (tr[u].l >= l && tr[u].r <= r)
	{
		change(u, x);
	}
	else
	{
		pushdown(u);
		int mid = tr[u].l + tr[u].r >> 1;
		if (l <= mid)
			modify(u << 1, l, r, x);
		if (r > mid)
			modify(u << 1 | 1, l, r, x);
		pushup(u);
	}
}
int query(int u, int l, int r)
{
	if (tr[u].l >= l && tr[u].r <= r)
	{
		return tr[u].xsum % mod;
	}
	pushdown(u);
	int mid = tr[u].l + tr[u].r >> 1;
	ll res = 0;
	if (l <= mid)
		res += query(u << 1, l, r);
	if (r > mid)
		res += query(u << 1 | 1, l, r);
	return res % mod;
}
signed main()
{
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> w[i];
	}
	build(1, 1, n);
	cin >> m;
	while (m--)
	{
		int op;
		int l, r, x;
		cin >> op;
		if (op == 0)
		{
			cin >> l >> r >> x;
			modify(1, l, r, x);
		}
		else
		{
			cin >> l >> r;
			cout << query(1, l, r) % mod << endl;
		}
	}
}

此处涉及公式

 注意在pushdown操作中,平方和需要的是区间加之前的区间和的值,所以需要先更新区间平方和,再更新区间和

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值