제출 #1131474

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1131474		VMaksimoski008	바이오칩 (IZhO12_biochips)	C++17	90 / 100	526 ms	403636 KiB

biochips

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 5;

int n, m, root, a[maxn], P[maxn], in[maxn], out[maxn], T=0, dp[maxn][505];
vector<int> G[maxn], E;

void dfs(int u) {
    in[u] = T++; E.push_back(u);
    for(int v : G[u]) dfs(v);
    out[u] = T;
}

int main() {
    cin >> n >> m;
    for(int i=0; i<=n; i++)
        for(int j=0; j<=m; j++) dp[i][j] = -1e9;

    for(int i=1; i<=n; i++) {
        cin >> P[i] >> a[i];
        if(!P[i]) root = i;
        else G[P[i]].push_back(i);
    }

    dfs(root); dp[n][0] = 0;
    for(int i=n-1; i>=0; i--) {
        for(int j=0; j<=m; j++) {
            dp[i][j] = dp[i+1][j];
            if(j) dp[i][j] = max(dp[i][j], dp[out[E[i]]][j-1] + a[E[i]]);
        }
    }
    cout << dp[0][m] << '\n';
}

• Subtask 1: 90 / 100