Submission #1279618

#	Time	Username	Problem	Language	Result	Execution time	Memory
1279618		not_amir	Magic Tree (CEOI19_magictree)	C++20	11 / 100	131 ms	104784 KiB

magictree

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

typedef long long ll;

struct vertex {
    vertex *vl = nullptr, *vr = nullptr;
    ll mxv = 0, add = 0;

    void extend(vertex *&v) {
        if (v == nullptr)
            v = new vertex();
    }

    void update(int tl, int tr, int l, int r, ll val, ll sum = 0) {
        if (l >= r)
            return;
        sum += add;
        if (l == tl && r == tr)
            mxv = max(mxv, val - sum);
        else {
            int tm = (tl + tr) / 2;
            extend(vl), vl->update(tl, tm, l, min(r, tm), val);
            extend(vr), vr->update(tm, tr, max(l, tm), r, val);
        }
    }

    ll get_max(int tl, int tr, int x, ll sum = 0) {
        sum += add;
        if (tl + 1 == tr)
            return sum + mxv;
        int tm = (tl + tr) / 2;
        return max(sum + mxv, x < tm ? vl ? vl->get_max(tl, tm, x) : 0 : vr ? vr->get_max(tm, tr, x) : 0);
    }
};

vertex* merge(vertex *v1, vertex *v2, ll m1 = 0, ll m2 = 0) {
    if (v1 == nullptr && v2 == nullptr) {
        vertex *v = new vertex();
        v->add = m1 + m2;
        return v;
    }
    if (v1 == nullptr) {
        v2->add += m1;
        return v2;
    }
    if (v2 == nullptr) {
        v1->add += m2;
        return v1;
    }
    vertex *v = new vertex();
    m1 = max(m1, v1->mxv), m2 = max(m2, v2->mxv);
    v->vl = merge(v1->vl, v2->vl, m1, m2);
    v->vr = merge(v1->vr, v2->vr, m1, m2);
    return v;
}

typedef pair<int, int> pii;

vector<vector<int>> G;
vector<pii> fruits;

int k;

vertex *dfs(int v) {
    vertex *ret = new vertex();
    for (int u : G[v])
        ret = merge(ret, dfs(u));
    auto [d, w] = fruits[v];
    if (d) {
        ll mxv = ret->get_max(0, k, d);
        ret->update(0, k, d, k, mxv + w);
    }
    return ret;
}

ll get_max(vertex *v, ll mxv = 0) {
    if (v == nullptr)
        return mxv;
    mxv = max(mxv, v->mxv);
    return v->add + max(get_max(v->vl, mxv), get_max(v->vr, mxv));
}

int main() {
    cin.tie(nullptr)->sync_with_stdio(false);
    int n, m;
    cin >> n >> m >> k;
    k++;
    G.resize(n + 1);
    fruits.resize(n + 1);
    for (int i = 2; i <= n; i++) {
        int p;
        cin >> p;
        G[p].push_back(i);
    }
    for (int i = 0; i < m; i++) {
        int v, d, w;
        cin >> v >> d >> w;
        fruits[v] = {d, w};
    }
    cout << get_max(dfs(1));
}

• Subtask 1: 0 / 6
• Subtask 2: 0 / 3
• Subtask 3: 11 / 11
• Subtask 4: 0 / 12
• Subtask 5: 0 / 16
• Subtask 6: 0 / 13
• Subtask 7: 0 / 22
• Subtask 8: 0 / 17