当对区间即有加操作,又有乘操作时:
//乘法满足分配率!!,所以乘懒标记可以“攻击”加懒标记
//策略:两个标记都安排
//当乘标记来临时,对自己和懒标记都乘//假设都没有向后延伸
//
//(特别好的分析:)
//当加标记来临时,正常加就好啦,因为乘已经对加处理啦
//
//两个一起来临呢,先乘!!!!!!!!!!!!!!!!!
//(乘已经对这部分加处理过了)
template
class ST//segment tree
{
struct node
{
T val;
T t1;
T t2;//乘
node(T v = 0) :val(v), t1(0), t2(1)//x2x3 == x6
{}
};
ll n = a.size();
vectora;
vectord;
public:
void build_tree(ll i, ll l, ll r)
{
if (l == r)
{
d[i].val = a[l] % MOD;
return;
}
ll mid = l + (r - l) / 2;
build_tree(i * 2, l, mid);
build_tree(i * 2 + 1, mid + 1, r);
d[i].val = d[i * 2].val + d[i * 2 + 1].val;
}
void spread(ll i, ll l, ll r, ll aiml, ll aimr)
{
//懒惰
ll mid = l + (r - l) / 2;
//if ((d[i].t1 != 0 || d[i].t2 != 1) && l != r)
//{
T t1 = d[i].t1, t2 = d[i].t2;
//先乘
d[i * 2].val = (d[i * 2].val * t2) % MOD;
d[i * 2].t1 = (d[i * 2].t1 * t2) % MOD;
d[i * 2].t2 = (d[i * 2].t2 * t2) % MOD;
d[i * 2 + 1].val = (d[i * 2 + 1].val * t2) % MOD;
d[i * 2 + 1].t1 = (d[i * 2 + 1].t1 * t2) % MOD;
d[i * 2 + 1].t2 = (d[i * 2 + 1].t2 * t2) % MOD;
//后加
d[i * 2].val = (d[i * 2].val + t1 * (mid - l + 1)) % MOD;
d[i * 2 + 1].val = (d[i * 2 + 1].val + t1 * (r - mid)) % MOD;
d[i * 2].t1 = (d[i * 2].t1 + t1) % MOD;
d[i * 2 + 1].t1 = (d[i * 2 + 1].t1 + t1) % MOD;
//复原
d[i].t1 = 0;
d[i].t2 = 1;
//}
///
}
ll getsum(ll l, ll r)
{
return _getsum(1, 1, n, l, r);
}
ll _getsum(ll i, ll l, ll r, ll aiml, ll aimr)
{
if (aiml <= l && r <= aimr)
return d[i].val;
ll mid = l + (r - l) / 2;
spread(i, l, r, aiml, aimr);
ll ret = 0;
if (aiml <= mid)
ret = (ret + _getsum(i * 2, l, mid, aiml, aimr)) % MOD;
if (aimr >= mid + 1)
ret = (ret + _getsum(i * 2 + 1, mid + 1, r, aiml, aimr)) % MOD;
return ret;
}
void update1(ll l, ll r, ll val)
{
_update1(1, 1, n, l, r, val);//加并挂标记
}
void _update1(ll i, ll l, ll r, ll aiml, ll aimr, T val)
{
if (aiml <= l && r <= aimr)
{
d[i].t1 = (d[i].t1 + val) % MOD;
d[i].val = (d[i].val + val * (r - l + 1)) % MOD;
return;
}
ll mid = l + (r - l) / 2;
spread(i, l, r, aiml, aimr);
if (aiml <= mid)
_update1(i * 2, l, mid, aiml, aimr, val);
if (aimr >= mid + 1)
_update1(i * 2 + 1, mid + 1, r, aiml, aimr, val);
d[i].val = (d[i * 2].val + d[i * 2 + 1].val) % MOD;
}
void update2(ll l, ll r, T val)
{
_update2(1, 1, n, l, r, val);//加并挂标记
}
void _update2(ll i, ll l, ll r, ll aiml, ll aimr,T val)
{
if (aiml <= l && r <= aimr)
{
d[i].val = (d[i].val * val) % MOD;
d[i].t1 = (d[i].t1 * val) % MOD;
d[i].t2 = (d[i].t2 * val) % MOD;
return;
}
ll mid = l + (r - l) / 2;
spread(i, l, r, aiml, aimr);
if (aiml <= mid)
_update2(i * 2, l, mid, aiml, aimr, val);
if (aimr >= mid + 1)
_update2(i * 2 + 1, mid + 1, r, aiml, aimr, val);
d[i].val = (d[i * 2].val + d[i * 2 + 1].val) % MOD;
}
ST(vectorarr)
{
a = arr;
n = a.size() - 1;
d = vector(pow(2, (ll)log2(n) + 1 + 1) - 1 + 1);
build_tree(1, 1, n);
}
};