洛谷 P1272 重建道路

题目描述

一场可怕的地震后，人们用N个牲口棚(1≤N≤150，编号1..N)重建了农夫John的牧场。由于人们没有时间建设多余的道路，所以现在从一个牲口棚到另一个牲口棚的道路是惟一的。因此，牧场运输系统可以被构建成一棵树。John想要知道另一次地震会造成多严重的破坏。有些道路一旦被毁坏，就会使一棵含有P(1≤P≤N)个牲口棚的子树和剩余的牲口棚分离，John想知道这些道路的最小数目。

输入输出格式
输入格式：

第1行：2个整数，N和P

第2..N行：每行2个整数I和J，表示节点I是节点J的父节点。

输出格式：

单独一行，包含一旦被破坏将分离出恰含P个节点的子树的道路的最小数目。

输入输出样例

输入样例#1：
11 6
1 2
1 3
1 4
1 5
2 6
2 7
2 8
4 9
4 10
4 11

输出样例#1：
2

说明

【样例解释】

如果道路1-4和1-5被破坏，含有节点（1，2，3，6，7，8）的子树将被分离出来

---

【分析】

能看出来是树形dp，然而做起来感觉和其他题套路不太一样，还是需要思考一波…思考了两晚终于屈服了。
（事实上我是借鉴记录里josen同学的思路，自己实在脑残…）
用 dp[i][j] 表示以i节点为根，截出含有j个点的连通子树所需要截的最少次数。
那么可以得到初始化 dp[i][1]=du[i] (du[i]为i的入边与出边的总和)，意思是只选i这一个节点，那么当然要把与它相连的边都截掉。
那么状态转移方程怎么得到呢
以样例为例子，节点1连接 2,3,4,5 。节点2连接 6,7,8。递归着进行动规之后我们可以得到 dp[2][3]=2 (截取1-2和2-8)
那么dp[1][4]=min(dp[1][4],dp[2][3]+dp[1][1]-2)
为啥要减2
因为dp[1][1]是删了一次 1-2 的结果,dp[2][3]也删了一次 1-2，但事实上得到dp[1][4]时 1-2是连通的，所以把这删的两次补上。
具体动规按照分组背包的循环顺序跑。

---

【代码】

//洛谷 P1272 重建道路 
#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#define ll long long
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=155;
vector <int> f[mxn];
int n,p;
int dp[mxn][mxn],du[mxn];
inline void dfs(int u)
{
    int i,j,k,v,x=f[u].size()-1;
    dp[u][1]=du[u];
    fo(i,0,x)
    {
        v=f[u][i];
        dfs(v);
        for(j=p;j>=2;j--)
          for(k=1;k<j;k++)
            dp[u][j]=min(dp[u][j],dp[u][k]+dp[v][j-k]-2);
    }
}
int main()
{
    int i,j,u,v,ans=1e8;
    scanf("%d%d",&n,&p);
    fo(i,0,n) fo(j,0,n) dp[i][j]=200; //初始化防止加法溢出 
    fo(i,0,n) dp[i][0]=0;
    fo(i,2,n)
    {
        scanf("%d%d",&u,&v);
        f[u].push_back(v);
        du[u]++;du[v]++;
    }
    dfs(1);
    fo(i,1,n)
      ans=min(ans,dp[i][p]);
    printf("%d\n",ans);
    return 0;
}