Submission #541210

# Submission time Handle Problem Language Result Execution time Memory
541210 2022-03-22T18:09:37 Z Olympia Sjeckanje (COCI21_sjeckanje) C++17
0 / 110
1 ms 340 KB
#include <vector>
#include <algorithm>
#include <iostream>
#include <set>
#include <cmath>
#include <map>
#include <random>
#include <cassert>
#include <ctime>
#include <cstdlib>
#include <limits.h>

using namespace std;
const int MOD = 1e9;

class Node {
public:
    pair<int64_t, int64_t> lval;
    pair<int64_t, int64_t> rval;
    int monotone;
    int64_t tot = 0;
};

void update(Node &x, int val) {
    if (x.monotone == -1) {
        x.tot = 0;
        x.rval = x.lval = {val, val};
        x.monotone = true;
    } else {
        x.lval.first += val;
        x.lval.second += val;
        x.rval.first += val;
        x.rval.second += val;
    }
}

Node merge(Node left, Node right) {
    if (left.monotone == -1) {
        return right;
    }
    if (right.monotone == -1) {
        return left;
    }
    Node ans;
    ans.tot = left.tot + right.tot;
    ans.lval = left.lval;
    ans.rval = right.rval;
    ans.monotone = false;
    pair<int64_t,int64_t> l = left.rval;
    pair<int64_t,int64_t> r = right.lval;
    if ((l.first <= l.second && l.second <= r.first && r.first <= r.second)
    || (l.first >= l.second && l.second >= r.first && r.first >= r.second)) {
        ans.tot -= abs(l.second - l.first);
        ans.tot -= abs(r.second - r.first);
        ans.tot += abs(l.first - r.second);
        if (left.monotone) {
            ans.lval = {l.first, r.second};
        }
        if (right.monotone) {
            ans.rval = {l.first, r.second};
        }
        if (left.monotone && right.monotone) {
            ans.monotone = true;
        }
    }
    return ans;
}

Node INF;

struct segmentTree {
    vector<Node> vec;
    vector<int64_t> addLater;

    void push(int v) {
        addLater[2 * v + 1] += addLater[v];
        addLater[2 * v] += addLater[v];
        update(vec[2 * v + 1], addLater[v]);
        update(vec[2 * v], addLater[v]);
        addLater[v] = 0;
    }

    void upd(int dum, int tl, int tr, int l, int r, int64_t val) {
        if (tr < l || tl > r) {
            return;
        }
        if (tl >= l && tr <= r) {
            addLater[dum] += val;
            update(vec[dum], val);
            return;
        }
        push(dum);
        int mid = (tl + tr) >> 1;
        upd(2 * dum, tl, mid, l, r, val);
        upd(2 * dum + 1, mid + 1, tr, l, r, val);
        vec[dum] = merge(vec[2 * dum], vec[2 * dum + 1]);
    }

    void upd(int l, int r, int val) {
        upd(1, 0, (int) vec.size() / 2 - 1, l, r, val);
    }

    Node get(int dum, int tl, int tr, int &l, int &r) {
        if (tl > r || tr < l) {
            return INF;
        }
        if (tl >= l && tr <= r) {
            return vec[dum];
        }
        push(dum);
        int tm = (tl + tr) >> 1;
        return merge(get(dum * 2, tl, tm, l, r), get(dum * 2 + 1, tm + 1, tr, l, r));
    }

    Node get(int l, int r) {
        return get(1, 0, (int) vec.size() / 2 - 1, l, r);
    }

    void resz(int n) {
        int sz = ((1 << (int) ceil(log2(n))));
        vec.assign(sz * 2, INF);
        addLater.resize(sz * 2);
    }

};

int main() {
    //freopen("inp.txt", "r", stdin);
    //freopen("haybales.out", "w", stdout);
    ios_base::sync_with_stdio(false);
    INF.monotone = -1;
    cin.tie(NULL);
    int N, D;
    cin >> N >> D;
    segmentTree st;
    st.resz(N);
    for (int i = 0; i < N; i++) {
        int x;
        cin >> x;
        st.upd(i, i, x);
    }
    while (D--) {
        int l, r, x;
        cin >> l >> r >> x;
        l--, r--;
        st.upd(l, r, x);
        cout << st.get(0, N - 1).tot << '\n';
    }
}
# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 340 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 340 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 340 KB Output isn't correct
2 Halted 0 ms 0 KB -