//#pragma GCC optimize("Ofast","O3")
//#pragma GCC target("avx2")
#include<bits/stdc++.h>
using namespace std;
const int N=200000;
int n,k;
vector<int> v(N);
vector<vector<int>> adj(N);
vector<vector<int>> dp(501,(vector<int>(N)));
vector<int> leaf(N);
void dfs(int k){
if(adj[k].size()==0){
leaf[k]=1;
}
for(int u:adj[k]){
dfs(u);
leaf[k]+=leaf[u];
for(int i=min(leaf[k],500);i>0;i--){
for(int j=1;j<=min(leaf[u],i);j++){
dp[i][k]=max(dp[i][k],dp[j][u]+dp[i-j][k]);
}
}
}
dp[1][k]=max(dp[1][k],v[k]);
return;
}
int main(){
ios_base::sync_with_stdio(false);
cin.tie(0);
int root=0;
cin>>n>>k;
for(int i=0;i<n;i++){
int p;cin>>p;p--;
if(p!=-1){
adj[p].push_back(i);
}
else{
root=i;
}
cin>>v[i];
}
dfs(root);
cout<<dp[k][root]<<endl;
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
155 ms |
400076 KB |
Output is correct |
2 |
Correct |
162 ms |
400112 KB |
Output is correct |
3 |
Correct |
156 ms |
400076 KB |
Output is correct |
4 |
Correct |
182 ms |
400380 KB |
Output is correct |
5 |
Correct |
191 ms |
400284 KB |
Output is correct |
6 |
Correct |
180 ms |
400352 KB |
Output is correct |
7 |
Correct |
675 ms |
401796 KB |
Output is correct |
8 |
Correct |
705 ms |
401872 KB |
Output is correct |
9 |
Execution timed out |
2100 ms |
401636 KB |
Time limit exceeded |
10 |
Execution timed out |
2079 ms |
401464 KB |
Time limit exceeded |