Codeforces Round #684 (Div. 2) E. Greedy Shopping 线段树+暴力剪枝

原题链接：https://codeforces.ml/contest/1440/problem/E

题意

有一个不递增的序列，有两个操作

1. x y 表示对[1, x]的元素取max(a[i], y)
2. x y 表示你有y元钱，对[x, n]的元素有序买东西，能买就买，输出购买的数量

分析

因为操作是对[1,x]取max，因此不论怎么操作数列仍然是不递增的，这是关键的题意转化，对于操作1的实现，我们记一个懒惰标记就可以轻松解决了。

对于操作2来说，我们要从左往右取，因此需要一些剪枝的技巧，我们记录一个区间最小值，如果区间最小值都大于你当前的钱，那么可以直接返回了。接着，如果钱大于区间和，那么就可以直接返回区间大小。这里可能有人疑惑怎么去记录区间和，因为有取max的操作，其实我们可以只在当前y大于区间最大值时取max，然后用最小值去剪枝，也可以保证时间复杂度。

Code

#include <bits/stdc++.h>
using namespace std;
//#define ACM_LOCAL
#define re register
#define fi first
#define se second
#define please_AC return 0
const int N = 2e5 + 10;
const int M = 1e6 + 10;
const int INF = 1e9;
const double eps = 1e-4;
const int MOD = 1e9+7;
typedef long long ll;
struct node {
    int l, r;
    int maxx, tag, minn;
    ll sum;
}t[N<<2];
int a[N];
void push_up(int u) {
    t[u].maxx = max(t[u<<1].maxx, t[u<<1|1].maxx);
    t[u].minn = min(t[u<<1].minn, t[u<<1|1].minn);
    t[u].sum = t[u<<1].sum + t[u<<1|1].sum;
}
void push_down(int u) {
    if (t[u].tag) {
        t[u<<1].tag = t[u].tag;
        t[u<<1|1].tag = t[u].tag;
        t[u<<1].maxx = t[u].tag;
        t[u<<1|1].maxx = t[u].tag;
        t[u<<1].minn = t[u].tag;
        t[u<<1|1].minn = t[u].tag;
        t[u<<1].sum = 1ll*t[u].tag * (t[u<<1].r - t[u<<1].l + 1);
        t[u<<1|1].sum = 1ll*t[u].tag * (t[u<<1|1].r - t[u<<1|1].l + 1);
        t[u].tag = 0;
    }
}
void build(int u, int l, int r) {
    t[u].l = l, t[u].r = r, t[u].tag = 0, t[u].minn = INF;
    if (l == r) {
        t[u].maxx = t[u].sum = t[u].minn = a[l];
        return;
    }
    int mid = (l + r) >> 1;
    build(u<<1, l, mid);
    build(u<<1|1, mid+1, r);
    push_up(u);
}
void modify(int u, int ql, int qr, int val) {
    if (t[u].minn >= val) return;
    if (ql <= t[u].l && qr >= t[u].r && val > t[u].maxx) {
        t[u].tag = val;
        t[u].maxx = t[u].minn = val;
        t[u].sum = 1ll*val * (t[u].r - t[u].l + 1);
        return;
    }
    push_down(u);
    int mid = (t[u].l + t[u].r) >> 1;
    if (ql <= mid) modify(u<<1, ql, qr, val);
    if (qr > mid) modify(u<<1|1, ql, qr, val);
    push_up(u);
}
int query(int u, int ql, int qr, int &mo) {
    if (t[u].minn > mo) return 0;
    if (ql <= t[u].l && qr >= t[u].r && mo >= t[u].sum) {
        mo -= t[u].sum;
        return t[u].r - t[u].l + 1;
    }
    push_down(u);
    int ans = 0;
    int mid = (t[u].l + t[u].r) >> 1;
    if (ql <= mid) ans += query(u<<1, ql, qr, mo);
    if (qr > mid) ans += query(u<<1|1, ql, qr, mo);
    return ans;
}
void solve() {
    int n, m; cin >> n >> m;
    for (int i = 1; i <= n; i++) cin >> a[i];
    build(1, 1, n);
    for (int i = 1; i <= m; i++) {
        int opt, x, y; cin >> opt >> x >> y;
        if (opt == 1) {
            modify(1, 1, x, y);
        } else {
            cout << query(1, x, n, y) << endl;
        }
    }
}

signed main() {
#ifdef ACM_LOCAL
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
#endif
    solve();
    return 0;
}