洛谷P2018 消息传递 树形DP

洛谷P2018 消息传递

树形DP
dp[ u ] 表示 u 节点 已经被传到，
然后将其字节点都传到所需要的最少时间

可知一个原则 一个树中的子树中如果同时开始传，那么最晚才能传到的，那他肯定最先开始传
因为本身需要的时间就大了，如果再晚一点开始，就要更晚才能结束了

也就是说从叶子向根传，将一个树中的子树 按照 他们直接传所需要的时间排序
优先传大的，再传小的

也就是一个点 只要负责好传到他们的儿子那些点就够了

 

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <vector>
using namespace std ; 

const int maxn = 1011,inf = 1e9 ; 
vector <int> f[maxn] ; 
int n,mi,x,sum ;
int dp[maxn],ans[maxn] ;  

inline void dfs(int u,int fa) 
{
    int cnt = 0,v ; 
    int tmp[maxn] ;  
    for(int i=0;i<f[ u ].size();i++) 
    {
        v = f[ u ][ i ] ; 
        if(v!=fa) 
        {
            dfs(v,u) ; 
            tmp[++cnt] = dp[ v ] ; 
        }
    }
    sort(tmp+1,tmp+cnt+1) ; 
    for(int i=1;i<=cnt;i++) 
        dp[ u ] = max(dp[ u ],tmp[ i ] + cnt -i+1) ;   //  其中 cnt-i+1  表示的是 传给之前的儿子  
                                                        //以及自身所需要的时间 
}

int main() 
{
    scanf("%d",&n) ; 
    for(int i=2;i<=n;i++) 
    {
        scanf("%d",&x) ; 
        f[x].push_back( i ) ; 
        f[i].push_back( x ) ; 
    }
    mi = inf ; 
    for(int i=1;i<=n;i++) 
    {
        for(int j=0;j<=n;j++) dp[ j ] = 0 ; 
        dfs( i,-1 ) ; 
        ans[ i ] = dp[ i ] ; 
        mi = min(mi,ans[ i ]) ; 
     }
    printf("%d\n",mi+1) ; 
    
    
    for(int i=1;i<=n;i++) 
        if(ans[ i ]==mi) printf("%d ",i) ; 
    return 0 ; 
}