제출 #1176740

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1176740		juliany2	Fireworks (APIO16_fireworks)	C++20	100 / 100	423 ms	99872 KiB

fireworks

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
#define all(x) (x).begin(), (x).end()

const int N = 3e5 + 7;
int n, m;
vector<array<ll, 2>> adj[N];
int lst[N];
// value of dp[0], points where slope changes
ll z[N];
multiset<ll> slope[N];

void dfs(int v) {
    for (auto &[u, w] : adj[v]) {
        dfs(u);

        if (adj[u].empty()) {
            slope[u] = {w, w};
            z[u] = w;
        } else {
            while (lst[u] > 1) {
                slope[u].erase(prev(slope[u].end()));
                lst[u]--;
            }

            vector<ll> v;
            while (v.size() < 2) {
                auto it = prev(slope[u].end());
                v.push_back(*it);
                slope[u].erase(it);
            }

            for (ll x : v)
                slope[u].insert(x + w);
            z[u] += w;
        }

        if (slope[u].size() > slope[v].size())
            slope[u].swap(slope[v]);

        z[v] += z[u];

        for (ll x : slope[u])
            slope[v].insert(x);

        lst[v]++;
    }
}

int main() {
    cin.tie(0)->sync_with_stdio(false);

    cin >> n >> m;

    for (int i = 2; i <= n + m; i++) {
        int p, w;
        cin >> p >> w;
        adj[p].push_back({i, w});
    }

    dfs(1);

    ll ans = z[1];

    while (lst[1] > 0) {
        slope[1].erase(prev(slope[1].end()));
        lst[1]--;
    }

    for (ll x : slope[1]) {
        ans -= x;
    }

    cout << ans << '\n';

    return 0;
}

• Subtask 1: 7 / 7
• Subtask 2: 19 / 19
• Subtask 3: 29 / 29
• Subtask 4: 45 / 45