Submission #939294

#	Time	Username	Problem	Language	Result	Execution time	Memory
939294		Alcabel	Fireworks (APIO16_fireworks)	C++17	19 / 100	18 ms	10740 KiB

fireworks

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

const int maxn = 3e5, maxc = 300;
long long dp[maxn][maxc + 1];
vector<int> g[maxn];
int par[maxn], c[maxn];

void print(int v) {
    for (int j = 0; j <= maxc; ++j) {
        cerr << dp[j] << ' ';
    }
    cerr << '\n';
}

long long tmp[maxc + 1];
void dfs(int v) {
    if (g[v].empty()) {
        for (int j = 0; j <= maxc; ++j) {
            dp[v][j] = abs(j - c[v]);
        }
        return;
    }
    for (const int& u : g[v]) {
        dfs(u);
        for (int j = 0; j <= maxc; ++j) {
            dp[v][j] += dp[u][j];
        }
    }
    // cerr << "v: " << v + 1 << " before:\n";
    // print(v);
    if (par[v] != -1) {
        for (int j = 0; j <= maxc; ++j) {
            tmp[j] = 1e18;
        }
        for (int j1 = 0; j1 <= maxc; ++j1) {
            for (int j2 = 0; j2 <= maxc - j1; ++j2) {
                tmp[j1 + j2] = min(tmp[j1 + j2], dp[v][j1] + abs(c[v] - j2));
            }
        }
        for (int j = 0; j <= maxc; ++j) {
            dp[v][j] = tmp[j];
        }
    }
    // cerr << "v: " << v + 1 << ", val: " << val[v] << ", slope: " << slope[v] << '\n';
    // print(v);
}

void solve() {
    int n, m;
    cin >> n >> m;
    par[0] = -1, c[0] = 0;
    for (int i = 1; i < n + m; ++i) {
        cin >> par[i] >> c[i];
        --par[i];
        g[par[i]].emplace_back(i);
    }
    dfs(0);
    long long ans = 1e18;
    for (int j = 0; j <= maxc; ++j) {
        ans = min(ans, dp[0][j]);
    }
    cout << ans << '\n';
}
 
signed main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
 
#ifdef LOCAL
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
    int T = 1;
    cin >> T;
    while (T--) {
        solve();
        cerr << "-----------\n";
        cout << "-----------\n";
    }
#else
    int T = 1;
    // cin >> T;
    while (T--) {
        solve();
    }
#endif
 
    return 0;
}

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