Submission #388029

#	Time	Username	Problem	Language	Result	Execution time	Memory
388029		b00n0rp	바이오칩 (IZhO12_biochips)	C++17	100 / 100	378 ms	400984 KiB

biochips

#include <bits/stdc++.h>
using namespace std;
 
#define FOR(i,a,b) for (int i = a; i < b; i++)
typedef vector<int> vi;
 
const int INF=1e9;
const int MAXN = 200005;
 
int n,m,root;
vi g[MAXN];
int par[MAXN],a[MAXN],st[MAXN],fin[MAXN];
int tim = 0;
int dp[MAXN][505];
 
void dfs(int u){
    st[u] = tim;
    for(auto v:g[u]) dfs(v);
    for(int i=1;i<=m;i++){
        dp[tim+1][i] = max(dp[tim][i],dp[st[u]][i-1]+a[u]);
    }
    tim++;
}   
 
signed main(){
    cin >> n >> m;
    for(int i=1;i<=n;i++){
        int p;
        cin >> p >> a[i];
        if(!p) root = i;
        else g[p].push_back(i);
    }
    for(int i=1;i<=m;i++) dp[0][i] = -INF;
    dfs(root);
    cout << dp[n][m] << "\n";
}
 
/*
int val[200001], m, t;
vector<int> g[200001];
int dp[200001][501], par[200001];
 
void dfs(int c)
{
    par[c] = t;
    for(int e : g[c]) dfs(e);
    for(int i=1;i<=m;i++)
        dp[t+1][i] = max(dp[t][i],dp[par[c]][i-1]+val[c]);
    t++;
}
 
int main()
{
    int n, r;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        int p;
        scanf("%d%d",&p,val+i);
        if(!p) r = i;
        else g[p].push_back(i);
    }
    for(int i=1;i<=m;i++) dp[0][i] = -1e9;
    dfs(r);
    printf("%d\n",dp[n][m]);
}
*/

• Subtask 1: 100 / 100