【数据结构】吉司机线段树

适用于:

  • 维护 区间历史最大/小值
  • 将区间内的所有数变成 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;
}
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Texcavator

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值