Submission #1218846

#	Time	Username	Problem	Language	Result	Execution time	Memory
1218846		countless	Progression (NOI20_progression)	C++20	48 / 100	856 ms	72408 KiB

Progression

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef long double ld;

#define sp <<" "<<
#define endl "\n"

const ll BAD = 0;
const ll INF = 1e18;

struct segtree {
    struct data {
        ll size;
        ll p, s;
        ll pref, suff, best;

        ll sum, padd, pset;

        data(ll size = -1, ll pref = -BAD, ll suff = -BAD, ll best = -BAD, ll p = -INF, ll s = -INF, ll sum = -BAD)
            : size(size), pref(pref), suff(suff), best(best), p(p), s(s), sum(sum), padd(0), pset(0) {;;}
    };

    ll l, r;
    data d;
    segtree *left, *right;

    inline data combine(const data &a, const data &b) {
        if (a.p == -INF or a.s == -INF) return b;
        if (b.p == -INF or b.s == -INF) return a;
        
        return data({
            a.size + b.size,
            (a.size == a.pref and a.p == b.p ? a.pref + b.pref : a.pref),
            (b.size == b.pref and a.s == b.s ? a.suff + b.suff : b.suff),
            max({a.best, b.best, (a.s == b.p ? a.suff + b.pref : 0)}),
            a.p,
            b.s,
            a.sum + b.sum
        });
    }

    inline void merge() {
        d = combine(left->d, right->d);
    }

    segtree(int l, int r, vector<ll> &a) : l(l), r(r) {
        if (l == r) {
            d = data(1, 1, 1, 1, a[l], a[l], a[l]);
            left = right = nullptr;
            return;
        }

        int m = (l+r) / 2;
        left = new segtree(l, m, a);
        right = new segtree(m+1, r, a);
        merge();
    }

    void propagate() {
        if (d.pset) {
            if (left) {
                left->d.p = d.pset;
                left->d.s = d.pset;
                left->d.sum = (left->r - left->l + 1) * d.pset;
                left->d.pset = d.pset;
                left->d.padd = 0;
            }

            if (right) {
                right->d.p = d.pset;
                right->d.s = d.pset;
                right->d.sum = (right->r - right->l + 1) * d.pset;
                right->d.pset = d.pset;
                right->d.padd = 0;
            }

            d.pset = 0;
        }

        if (d.padd) {
            if (left) {
                left->d.p += d.padd;
                left->d.s += d.padd;
                left->d.padd += d.padd;
            }

            if (right) {
                right->d.p += d.padd;
                right->d.s += d.padd;
                right->d.padd += d.padd;
            }

            d.padd = 0;
        }
    }

    void add(int ql, int qr, ll upd) {
        propagate();

        if (ql > r or qr < l) return;

        if (ql <= l and r <= qr) {
            d.p += upd;
            d.s += upd;
            d.sum += (r - l + 1) * upd;
            d.padd += upd;
            propagate();
            return;
        }

        left->add(ql, qr, upd);
        right->add(ql, qr, upd);
        merge();
    }

    void set(int ql, int qr, ll upd) {
        propagate();

        if (ql > r or qr < l) return;

        if (ql <= l and r <= qr) {
            d.p = upd;
            d.s = upd;
            d.sum = (r - l + 1) * upd;
            d.pset = upd;
            propagate();
            return;
        }

        left->add(ql, qr, upd);
        right->add(ql, qr, upd);
        merge();
    }

    data query(int ql, int qr) {
        propagate();

        if (ql > r or qr < l) return data();

        if (ql <= l and r <= qr) return d;

        return combine(left->query(ql, qr), right->query(ql, qr));
    }
};

void solve() {
    int n, q; cin >> n >> q;
    vector<ll> a(n), d1;
    for (int i = 0; i < n; i++) cin >> a[i];
    
    auto derive = [&](vector<ll> &a) -> vector<ll> {
        vector<ll> res(n);
        for (int i = 1; i < n; i++) {
            res[i] = a[i] - a[i-1];
        }
        return res;
    };

    // create diff array
    d1 = derive(a);
    d1[0] = -INF;

    // build segtree
    segtree st(0, n-1, d1);

    // answer queries
    int type, l, r, s, c;
    while (q--) {
        cin >> type;

        if (type == 1) {
            cin >> l >> r >> s >> c; l--, r--;
            ll sz = (r - l + 1);

            st.add(l, l, s);
            st.add(l+1, r, c);
            if (r+1 < n) st.add(r+1, r+1, -(s + (sz - 1) * c));
        } else

        if (type == 2) {
            cin >> l >> r >> s >> c; l--, r--;


            assert(0);
        } else 

        if (type == 3) {
            cin >> l >> r; l--, r--;
            ll sz = (r - l + 1);
            cout << min(sz, max(1LL, st.query(l+1, r).best) + 1) << endl;
        }
    }
}

signed main() {
    cin.tie(0);
    ios_base::sync_with_stdio(false);

    int t = 1;
    // cin >> t;
    while (t--)
        solve();
}

• Subtask 1: 0 / 9
• Subtask 2: 0 / 15
• Subtask 3: 35 / 35
• Subtask 4: 0 / 11
• Subtask 5: 13 / 13
• Subtask 6: 0 / 17