这道题是树结构上的DP,明天也会继续做这类题。
思路:建树,记录每个节点的直接儿子节点,然后dfs到叶子节点,然后计算它的父节点的d[i](d[i]表示第i个中级员工(节点)的下面至少需要多少个工人签名。这里需要排序一下)。
我开始想法是建树然后后序遍历,每次更新叶子节点上的节点,然而排序起来很不方便。对比之下,dfs写出来的代码就漂亮多了。看来碰到树结构问题的时候可以多往dfs方向去想。
#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;
int dp(int u,vector<int> v[]){///vector传参
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;
}