HDU 1561 树形DP(入门)

题目链接:  HDU 1561 The more, The Better

 

#include <iostream>

#include <cstdio>

#include <cmath>

#include <cstdlib>

#include <string>

#include <cstring>

#include <algorithm>

#include <iomanip>

#include <vector>



using namespace std;

const int inf = 0x7FFFFFFF;

const int maxn = 1000000;

vector<int>V[205];



int n,m;

int dp[205][205];

bool vis[205];



void Init(){

    for(int i=0;i<=n;++i) 

        V[i].clear();

    memset(dp,0,sizeof(dp));

    memset(vis,false,sizeof(vis));

    for(int i=0; i<=n; ++i)

        for(int j=2; j<=m; ++j)

            dp[i][j]=-1;

}



void DFS(int cnt){

    vis[cnt]=true;

    int len=V[cnt].size();

    for(int i=0;i<len;++i){

        int next=V[cnt][i];

        if(!vis[next]) DFS(next);

        for(int j=m;j>=2;--j)

            for(int k=1;k<j;++k)

                if(dp[next][j-k]!=-1&&dp[cnt][k]!=-1)

                    dp[cnt][j]=max(dp[cnt][j],dp[next][j-k]+dp[cnt][k]);

    }

}



int main(){

    while(~scanf("%d%d",&n,&m),n+m){

        Init();

        for(int i=1;i<=n;++i){

            int a,b;

            scanf("%d%d",&a,&b);

            dp[i][1]=b;

            V[a].push_back(i);  /// a的子结点

        }

        ++m;

        DFS(0);

        printf("%d\n",dp[0][m]);

    }

    return 0;

}