POJ 1947 - Rebuilding Roads 트리 DP(일반 가방 이동)...

#include<iostream>

#include<stdio.h>

#include<string.h>

#include<set>

#include<ctime>

#include<algorithm>

#include<queue>

#include<cmath>

#include<map>

#define oo 100000007

#define ll long long

#define pi acos(-1.0)

#define MAXN 155

using namespace std;

vector<int> Tree[MAXN];

int dp[MAXN][MAXN],N,P,ans;

bool root[MAXN];

int dfs(int x)

{

      int i,j,y,m=Tree[x].size(),num=1,t,update;  

      for (i=0;i<=P;i++) dp[x][i]=oo;

      dp[x][1]=0;

      for (i=0;i<m;i++)

      {

             y=Tree[x][i];

             num+=dfs(y); 

             for (t=P;t>=1;t--)

             {

                    update=dp[x][t]+1;

                    for (j=1;j<=t;j++)

                       update=min(update,dp[x][t-j]+dp[y][j]);

                    dp[x][t]=update;

             }  //       

      }

      t=0;

      if (!root[x]) t++;

      if (dp[x][P]!=-1) ans=min(dp[x][P]+t,ans);

      return num;

}

int main()

{

      int i; 

      while (~scanf("%d%d",&N,&P))

      {

            for (i=1;i<=N;i++) Tree[i].clear();

            memset(root,true,sizeof(root));

            for (i=1;i<N;i++)

            {

                   int x,y;

                   scanf("%d%d",&x,&y);

                   Tree[x].push_back(y);

                   root[y]=false;

            }

            for (i=1;i<=N;i++)

               if (root[i]) break;

            ans=oo;

            dfs(i);

            printf("%d
",ans);

      }

      return 0;

}