# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
41120 | wzy | Biochips (IZhO12_biochips) | C++11 | 2065 ms | 415008 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <vector>
#include <algorithm>
#include <iostream>
#include <utility>
using namespace std;
#define eps (int) 1e9
#define pb push_back
int root , rv[200005], t = 0 , maxrv = 0 , dp[200005][505] , c[200005] , n , m , st[200005] , ed[200005];
vector<int> adj[200005];
void dfs(int x , int y){
st[x] = ++t;
rv[st[x]] = x;
maxrv = max(maxrv , st[x]);
for(int i = 0 ; i < adj[x].size() ; i++){
int v = adj[x][i];
if(v == y) continue;
dfs(v,x);
}
ed[x] = t;
return;
}
int solve(int i, int j){
if(i == ed[rv[maxrv]] + 1 && j < m) return -eps;
if(dp[i][j] != -1) return dp[i][j];
if(j == m) return 0;
if(i > maxrv) return -eps;
dp[i][j] = 0;
return dp[i][j] = max(solve(i + 1 , j) , solve(ed[rv[i]] + 1 , j + 1) + c[rv[i]]);
}
int32_t main(){
scanf("%d%d" , &n , &m);
memset(dp, -1 , sizeof dp);
for(int i = 0 ; i <n; i++){
int x , y;
scanf("%d%d" , &x, &y);
c[i] = y;
if(x == 0){
root = i;
continue;
}
else{
adj[i].pb(x-1);
adj[x-1].pb(i);
}
}
dfs(root, root);
int qq = solve(st[root] , 0);
printf("%d\n" , qq);
}
Compilation message (stderr)
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
Fetching results... |