oj.uz

답안 #388022

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
388022 2021-04-09T18:40:18 Z b00n0rp 바이오칩 (IZhO12_biochips) C++17
100 / 100
 344 ms 400964 KB 
#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 val[MAXN],par[MAXN];
int t = 0;
int dp[MAXN][505];

void dfs(int u){
    par[u] = t;
    for(auto v:g[u]) dfs(v);
    for(int i=1;i<=m;i++){
        dp[t+1][i] = max(dp[t][i],dp[par[u]][i-1]+val[u]);
    }
    t++;
}   
 
signed main(){
    cin >> n >> m;
    for(int i=1;i<=n;i++){
        int p;
        cin >> p >> val[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.0 / 100.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 4940 KB Output is correct
2 Correct 3 ms 4940 KB Output is correct
3 Correct 3 ms 5196 KB Output is correct
4 Correct 15 ms 22748 KB Output is correct
5 Correct 17 ms 24908 KB Output is correct
6 Correct 17 ms 24908 KB Output is correct
7 Correct 251 ms 298692 KB Output is correct
8 Correct 264 ms 298780 KB Output is correct
9 Correct 308 ms 363280 KB Output is correct
10 Correct 344 ms 400964 KB Output is correct