这题,细思极恐,这么短的代码就实现了,参考了别人的代码写出来的,感觉比紫书上的代码好懂,看来要好好训练dfs+dp的这种解法。
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#include<stdlib.h>
#include<iostream>
#include<cmath>
using namespace std;
const int maxn=1e5+50;
int n,t,no,ans;
double percent;
//vector<int> v[maxn];///v[u][v] u号连着编号为v儿子
int dp(int u,vector<int> v[]){
if(v[u].size()==0)
return 1;
vector<int> d;
for(int i=0;i<v[u].size();i++){
d.push_back(dp(v[u][i],v));
}
sort(d.begin(),d.end());
int num=ceil(v[u].size()*percent);
ans=0;
for(int i=0;i<num;i++){
ans+=d[i];
}
return ans;
}
int main(){
while(scanf("%d%d",&n,&t)!=EOF){
if(n==0&&t==0)
break;
vector<int> v[maxn];///v[u][v] u号连着编号为v儿子
percent=(1.0*t)/100;
for(int i=1;i<=n;i++){
scanf("%d",&no);
v[no].push_back(i);
}
printf("%d\n",dp(0,v));
}
return 0;
}