Submission #1279578

#	Time	Username	Problem	Language	Result	Execution time	Memory
1279578		not_amir	Magic Tree (CEOI19_magictree)	C++20	26 / 100	148 ms	100576 KiB

magictree

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

typedef long long ll;

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

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

    void update(int tl, int tr, int x, ll val) {
        mxv = max(mxv, val);
        if (tl + 1 == tr)
            return;
        int tm = (tl + tr) / 2;
        if (x < tm)
            extend(vl), vl->update(tl, tm, x, val);
        else
            extend(vr), vr->update(tm, tr, x, val);
    }

    ll query(int tl, int tr, int l, int r) {
        if (l >= r)
            return 0;
        if (l == tl && r == tr)
            return mxv;
        int tm = (tl + tr) / 2;
        return max(
            vl ? vl->query(tl, tm, l, min(r, tm)) : 0,
            vr ? vr->query(tm, tr, max(l, tm), r) : 0);
    }
};

vertex* merge(vertex *v1, vertex *v2) {
    vertex *v = new vertex();
    v->mxv = v1->mxv + v2->mxv;
    int cntl = (v1->vl == nullptr ? 0 : 1) + (v2->vl == nullptr ? 0 : 1);
    int cntr = (v1->vr == nullptr ? 0 : 1) + (v2->vr == nullptr ? 0 : 1);
    if (cntl == 1) {
        if (v1->vl)
            v->vl = v1->vl;
        else
            v->vl = v2->vl;
    }
    if (cntl == 2)
        v->vl = merge(v1->vl, v2->vl);
    if (cntr == 1) {
        if (v1->vr)
            v->vr = v1->vr;
        else
            v->vr = v2->vr;
    }
    if (cntr == 2)
        v->vr = merge(v1->vr, v2->vr);
    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->query(0, k, 0, d + 1);
        ret->update(0, k, d, mxv + w);
    }
    return ret;
}

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 << dfs(1)->mxv;
}

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