适用于:
- 维护 区间历史最大/小值
- 将区间内的所有数变成 m a x ( a i , x ) max(a_i,\ x) max(ai, x)(或者 m i n min min)
模板题:P6242 【模板】线段树 3(区间最值操作、区间历史最值)
#include <bits/stdc++.h>
#include <bits/extc++.h>
using namespace std;
using namespace __gnu_pbds;
#define int long long
#define double long double
using i64 = long long;
typedef pair<int, int> PII;
typedef pair<int, char> PIC;
typedef pair<double, double> PDD;
typedef pair<PII, int> PIII;
typedef pair<int, pair<int, bool>> PIIB;
const int N = 5e5 + 10;
const int maxn = 1e6;
const int MAXN = 100;
const int mod = 998244353;
const int mod1 = 954169327;
const int mod2 = 906097321;
const int INF = 0x3f3f3f3f3f3f3f3f;
struct node {
int l, r;
int sum; // 区间和
int maxa, maxb; // 当前区间最大值和历史区间最大值
int cnt; // 当前区间最大值的个数
int se; // 当前区间次大值
int add1, add2, add3, add4; // 分别表示当前区间最大值的懒标记 当前区间非最大值的懒标记 add1的最大值 add2的最大值
} tr[N * 4];
int a[N];
void pushup(node& u, node& lt, node& rt)
{
u.sum = lt.sum + rt.sum;
u.maxa = max(lt.maxa, rt.maxa);
u.maxb = max(lt.maxb, rt.maxb);
if (lt.maxa == rt.maxa) // 左右节点最大值一样大
{
u.se = max(lt.se, rt.se);
u.cnt = lt.cnt + rt.cnt;
}
else if (lt.maxa > rt.maxa) // 左结点最大值比右结点大
{
u.se = max(lt.se, rt.maxa);
u.cnt = lt.cnt;
}
else // 右结点最大值比左结点大
{
u.se = max(lt.maxa, rt.se);
u.cnt = rt.cnt;
}
}
void pushup(int u)
{
pushup(tr[u], tr[u << 1], tr[u << 1 | 1]);
}
void build(int u, int l, int r)
{
tr[u] = {l, r};
if (l == r)
{
tr[u].sum = tr[u].maxa = tr[u].maxb = a[l];
tr[u].cnt = 1, tr[u].se = -INF;
return;
}
int mid = l + r >> 1;
build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
pushup(u);
}
void change(int u, int add1, int add2, int add3, int add4)
{
tr[u].sum += 1ll * add1 * tr[u].cnt + 1ll * add2 * (tr[u].r - tr[u].l + 1 - tr[u].cnt);
tr[u].maxb = max(tr[u].maxb, tr[u].maxa + add3);
tr[u].maxa += add1;
if (tr[u].maxb != -INF) tr[u].se += add2;
tr[u].add3 = max(tr[u].add3, tr[u].add1 + add3);
tr[u].add4 = max(tr[u].add4, tr[u].add2 + add4);
tr[u].add1 += add1, tr[u].add2 += add2;
}
void pushdown(int u)
{
int maxx = max(tr[u << 1].maxa, tr[u << 1 | 1].maxa);
// 修改左结点
if (tr[u << 1].maxa == maxx) change(u << 1, tr[u].add1, tr[u].add2, tr[u].add3, tr[u].add4);
else change(u << 1, tr[u].add2, tr[u].add2, tr[u].add4, tr[u].add4);
// 修改右结点
if (tr[u << 1 | 1].maxa == maxx) change(u << 1 | 1, tr[u].add1, tr[u].add2, tr[u].add3, tr[u].add4);
else change(u << 1 | 1, tr[u].add2, tr[u].add2, tr[u].add4, tr[u].add4);
// 清空懒标记
tr[u].add1 = tr[u].add2 = tr[u].add3 = tr[u].add4 = 0;
}
// 当前结点u 把[l,r]中元素全部 + x
void update_add(int u, int l, int r, int x)
{
if (tr[u].l >= l && tr[u].r <= r)
{
tr[u].sum += 1ll * x * (tr[u].r - tr[u].l + 1);
tr[u].maxa += x;
tr[u].maxb = max(tr[u].maxb, tr[u].maxa);
if (tr[u].se != -INF) tr[u].se += x;
tr[u].add1 += x, tr[u].add2 += x;
tr[u].add3 = max(tr[u].add3, tr[u].add1);
tr[u].add4 = max(tr[u].add4, tr[u].add2);
}
else
{
pushdown(u);
int mid = tr[u].l + tr[u].r >> 1;
if (l <= mid) update_add(u << 1, l, r, x);
if (r > mid) update_add(u << 1 | 1, l, r, x);
pushup(u);
}
}
// 当前结点u 把[l,r]中元素全部变为min(x,a[i])
void update_min(int u, int l, int r, int x)
{
if (x >= tr[u].maxa) return;
if (tr[u].l >= l && tr[u].r <= r && tr[u].se < x)
{
int k = tr[u].maxa - x;
tr[u].sum -= 1ll * tr[u].cnt * k;
tr[u].maxa = x, tr[u].add1 -= k;
}
else
{
pushdown(u);
int mid = tr[u].l + tr[u].r >> 1;
if (l <= mid) update_min(u << 1, l, r, x);
if (r > mid) update_min(u << 1 | 1, l, r, x);
pushup(u);
}
}
// 区间查询
node query(int u, int l, int r)
{
if (tr[u].l >= l && tr[u].r <= r) return tr[u];
pushdown(u);
int mid = tr[u].l + tr[u].r >> 1;
if (r <= mid) return query(u << 1, l, r);
else if (l > mid) return query(u << 1 | 1, l, r);
else
{
node res;
auto lt = query(u << 1, l, mid);
auto rt = query(u << 1 | 1, mid + 1, r);
pushup(res, lt, rt);
return res;
}
}
void solve()
{
int n, m; cin >> n >> m;
for (int i = 1; i <= n; i ++ ) cin >> a[i];
build(1, 1, n);
while (m -- )
{
int op, l, r; cin >> op >> l >> r;
if (op == 1)
{
int x; cin >> x;
update_add(1, l, r, x);
}
else if (op == 2)
{
int x; cin >> x;
update_min(1, l, r, x);
}
else
{
auto res = query(1, l, r);
if (op == 3) cout << res.sum << '\n';
else if (op == 4) cout << res.maxa << '\n';
else if (op == 5) cout << res.maxb << '\n';
}
}
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
int t = 1;
// cin >> t;
while (t--)
{
solve();
}
return 0;
}