题目链接:http://poj.org/problem?id=1947
题目大意及思路:求从一棵树上切下p个结点的子树最少需要切的边的条数,做法是树形dp加背包。
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define Max 110
int max(int a,int b)
{
return a>b?a:b;
}
int min(int a,int b)
{
return a<b?a:b;
}
int n,m,eid=0;
int p[160];
int dp[160][160],pre[160];
struct node
{
int to,next;
}e[210];
inline void addedge(int u,int v)
{
e[eid].to=v;
e[eid].next=p[u];
p[u]=eid++;
}
void dfs(int u)
{
// printf("%d\n",u);
int i,j,v,k,num=0;
for(i=p[u];i!=-1;i=e[i].next)
{
num++;
dfs(e[i].to);
}
for(i=0;i<=m;i++)
dp[u][i]=inf;
if(pre[u]!=0)
dp[u][1]=num+1;
else
dp[u][1]=num;
for(i=p[u];i!=-1;i=e[i].next)
{
v=e[i].to;
for(j=m;j>=0;j--)
{
if(dp[u][j]==inf)
continue;
for(k=1;k<=m&&j+k<=m;k++)
{
if(dp[v][k]!=inf)
{
dp[u][j+k]=min(dp[u][j+k],dp[v][k]+dp[u][j]-2);
}
}
}
}
}
int main()
{
int i,j,k,u,v;
while(scanf("%d%d",&n,&m)!=EOF)
{
eid=0;
memset(pre,0,sizeof(pre));
memset(p,-1,sizeof(p));
for(i=1;i<n;i++)
{
scanf("%d%d",&u,&v);
addedge(u,v);
pre[v]=u;
}
for(i=0;i<=n;i++)
for(j=0;j<=m;j++)
dp[i][j]=inf;
for(i=1;i<=n;i++)
{
if(pre[i]==0)
{
dfs(i);
break;
}
}
int ans=inf;
for(i=1;i<=n;i++)
{
ans=min(ans,dp[i][m]);
}
printf("%d\n",ans);
}
}