Submission #1128151

#	Time	Username	Problem	Language	Result	Execution time	Memory
1128151		_callmelucian	Magic Tree (CEOI19_magictree)	C++17	100 / 100	101 ms	34632 KiB

magictree

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

typedef long long ll;
typedef long double ld;
typedef pair<ll,ll> pl;
typedef pair<int,int> pii;
typedef tuple<int,int,int> tpl;

#define all(a) a.begin(), a.end()
#define filter(a) a.erase(unique(all(a)), a.end())

const int mn = 1e5 + 5;

struct point {
    mutable int x;
    mutable ll delta;

    point() : x(0), delta(0) {}
    point (int x, ll delta) : x(x), delta(delta) {}

    bool operator < (const point &o) const {
        return (x == o.x ? delta < o.delta : x < o.x);
    }
};

struct stair : multiset<point> {
    auto addPoint (int x, ll delta) {
        auto cur = emplace(x, delta), nxt = next(cur);
        while (nxt != end() && cur->x == nxt->x)
            cur->delta += nxt->delta, nxt = erase(nxt);
        return cur;
    }

    void incrPoint (int x, ll delta) {
        auto cur = addPoint(x, delta), nxt = next(cur);
        while (nxt != end()) {
            if (delta >= nxt->delta) delta -= nxt->delta, nxt = erase(nxt);
            else return nxt->delta -= delta, void();
        }
    }

    void merge (stair &o) {
        for (auto it : o) addPoint(it.x, it.delta);
        o.clear();
    }
} dp[mn];

int day[mn], weight[mn];
vector<int> adj[mn];

void dfs (int u, int p) {
    for (int v : adj[u])
        if (v != p) dfs(v, u);

    if (weight[u]) dp[u].incrPoint(day[u], weight[u]);
    if (u != p) {
        if (dp[u].size() > dp[p].size()) swap(dp[u], dp[p]);
        dp[p].merge(dp[u]);
    }
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);

    int n, m, k; cin >> n >> m >> k;
    for (int i = 1; i <= n; i++) day[i] = k;

    for (int i = 2; i <= n; i++) {
        int p; cin >> p;
        adj[p].push_back(i);
    }
    for (int i = 0; i < m; i++) {
        int v; cin >> v;
        cin >> day[v] >> weight[v];
    }
    dfs(1, 1);

    ll ans = 0;
    for (point item : dp[1]) ans += item.delta;
    cout << ans << "\n";

    return 0;
}

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