Submission #541238

# Submission time Handle Problem Language Result Execution time Memory
541238 2022-03-22T20:02:29 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:
    int64_t dp[2][2] = {{0, 0}, {0, 0}};
    int64_t left = 0;
    int64_t right = 0;
    int64_t tl, tr;
};

Node merge (Node x, Node y) {
    Node ans;
    ans.left = x.left;
    ans.right = y.right;
    ans.tl = x.tl;
    ans.tr = y.tr;
    for (int i = 0; i < 2; i++) {
        for (int j = 0; j < 2; j++) {
            ans.dp[i][j] = max(ans.dp[i][j], x.dp[i][1] + y.dp[1][j]);
            ans.dp[i][j] = max(ans.dp[i][j], x.dp[i][0] + abs(x.right - y.left) + y.dp[0][j]);
        }
    }
    if (x.tr - x.tl + y.tr - y.tl == 0) {
        ans.dp[0][1] = ans.dp[1][0] = ans.dp[0][0] = 0;
    }
    if (x.tr - x.tl + y.tr - y.tl == 1) {
        ans.dp[0][0] = 0;
    }
    return ans;
}

void update(Node &x, int val) {
    x.left += val, x.right += val;
}

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);
    }

    void init (int dum, int tl, int tr) {
        if (dum >= vec.size()) {
            return;
        }
        vec[dum].tl = tl, vec[dum].tr = min(tr, N - 1);
        int mid = (tl + tr) >> 1;
        init(2 * dum, tl, mid);
        init(2 * dum + 1, mid + 1, tr);
    }

    int N;

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

};
segmentTree st;
int N;

int main() {
    //freopen("inp.txt", "r", stdin);
    //freopen("haybales.out", "w", stdout);
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int D;
    cin >> N >> D;
    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 << max(max(st.vec[1].dp[0][1], st.vec[1].dp[1][1]), max(st.vec[1].dp[1][0], st.vec[1].dp[0][0])) << '\n';
    }
}

Compilation message

Main.cpp: In member function 'void segmentTree::init(int, int, int)':
Main.cpp:84:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<Node>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   84 |         if (dum >= vec.size()) {
      |             ~~~~^~~~~~~~~~~~~
# 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 -