题目大意:
有一棵树,我们去掉任意边,去掉边后剩下的节点的数等于p。问我们最少去掉多少的边。
解题思路:
这种题目是典型的树形DP,每一个节点有点类似于背包。背包的特点就是:到达某个地方选和不选,这里同样的我们也是到达某个节点后选和不选。怎么选?选什么呢?
这题是这样的:dp[u][j]= min (dp[u][j],dp[u][j-k]+dp[v][k])
其中:u代表我们到达了树节点u,j代表我们去掉边后剩下j个节点,dp[u][j]就是最少的删除边数,v是和u邻接的节点。这种树形DP,需要填充DP表的时候需要不断DFS到达叶子结点,这里的dp表边界条件是:
dp[u][1]=deg[u] ,其中deg表示节点的度数。
树形DP的模板是这样的:
void dfs(int u){
dp[][]=... //表示边界条件
for(auto nx){
dfs(nx);
for(int k=max;k>=1;k--) //注意这里必须从大到小!这是树形DP的特点。
for(int i=1;i<k;i++){
dp[u][k]= ...//转移
}
}
}
AC代码
#include <bits/stdc++.h>
using namespace std;
const int MAXN=160;
vector<vector<int>> gra(MAXN);
int dp[MAXN][MAXN];
int deg[MAXN];
int n,p;
const int INF=1e9;
int res=INF;
void dfs(int u){
dp[u][1]=deg[u];
for(int i=0;i<(int)gra[u].size();i++){
int nx=gra[u][i];
dfs(nx);
for(int j=p;j>=1;j--){
for(int k=1;k<j;k++){
dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[nx][k]-2);
}
}
}
res=min(res,dp[u][p]);
}
int main(){
cin>>n>>p;
set<pair<int,int>> ms;
for(int i=0;i<MAXN;i++)deg[i]=0;
for(int i=0;i<n-1;i++){
int a,b;cin>>a>>b;
if(a>b)swap(a,b);
if(ms.count({a,b}))continue;
else ms.insert({a,b});
a-=1;
b-=1;
gra[a].emplace_back(b);
deg[a]++;
deg[b]++;
}
for(int i=0;i<MAXN;i++)
for(int j=0;j<MAXN;j++)
dp[i][j]=INF;
dfs(0);
cout<<res<<endl;
return 0;
}