#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";
}
# |
Verdict |
Execution time |
Memory |
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 |
16 ms |
22732 KB |
Output is correct |
5 |
Correct |
17 ms |
24964 KB |
Output is correct |
6 |
Correct |
17 ms |
24908 KB |
Output is correct |
7 |
Correct |
262 ms |
298828 KB |
Output is correct |
8 |
Correct |
260 ms |
298764 KB |
Output is correct |
9 |
Correct |
297 ms |
363304 KB |
Output is correct |
10 |
Correct |
355 ms |
400896 KB |
Output is correct |