UVa 12186 树形dp

题意  分析   白皮书 P282  例题9-12

AC代码

 #include <stdio.h>
 #include <math.h>
 #include <string.h>
 #include <stdlib.h>
 #include <iostream>
 #include <sstream>
 #include <algorithm>
 #include <string>
 #include <queue>
 #include <vector>
 using namespace std;
 const int maxn= 1e5+;
 const int maxm= 1e4+;
 const int inf = 0x3f3f3f3f;
 typedef long long ll;
 int n,t;
 vector<int> sons[maxn];
 int dp(int a)　　
 {
     if(sons[a].empty())
         return ;
     int k=sons[a].size();
     vector<int> d;
     for(int i=; i<k; i++)
         d.push_back(dp(sons[a][i]));　　　　　　//直接递归下去
     sort(d.begin(),d.end());
     int c=(k*t-)/+;
     int ans=;
     for(int i=; i<c; i++)
         ans+=d[i];
     return ans;
 }
 int main()
 {
     while(scanf("%d %d",&n,&t)!=EOF)
     {
         if(n==&&t==)
             break;
         for(int i=; i<=n; i++)    //多组输入 清空vector 注意n也要清空
             sons[i].clear();
         int m;
         for(int i=; i<=n; i++)
         {　　　　　　　　　　　　　　　　
             scanf("%d",&m);　　　　　　　//输入第i个点的父节点是m 将i存到m的儿子数组里
             sons[m].push_back(i);
         }
         printf("%d\n",dp());
     }
     return ;
 }