POJ 1947 树形DP

题目链接

题意：

求树中最少的去除边数能够分割出一个节点数为k的子树。

思路：

定义dp[i][j]表示以i为根的子树分割出一个节点数为j的子树的最少去除边数。

状态转移：dp[u][j] = min ( dp[u][j] , dp[u][j-k]+dp[v][k]-1 );

u：当前树树根

v：子树树根

j： 当前树节点数

k：子树节点数

答案要+1，是因为初始定义dp[u][1]时统一使用了indeg[u]-1。这对非根来说是正好的但是非根最后独立出来需要与父节点断开需要多断开一条边，对于根节点来说初始答案就少1需要+1拯救一下。

C++代码：

#include<map>
#include<set>
#include<stack>
#include<cmath>
#include<queue>
#include<vector>
#include<string>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 210;
const int maxm = 410;
const int inf  = 0x3f3f3f3f;

int n,m,tol,head[maxn];
struct edge
{
    int to,next;
}es[maxm];

void addedge( int u , int v )
{
    es[tol].to = v;
    es[tol].next = head[u];
    head[u] = tol++;
}

int indeg[maxn],dp[maxn][maxn],sizes[maxn],ans;

void dfs( int u , int f )
{
    dp[u][1] = indeg[u]-1;
    sizes[u] = 1;
    for ( int i=head[u] ; i!=-1 ; i=es[i].next )
    {
        int v = es[i].to;
        if ( v!=f )
        {
            dfs( v , u );
            sizes[u] += sizes[v];
            for ( int j=sizes[u] ; j>1 ; j-- )
                for ( int k=1 ; k<j&&k<=sizes[v] ; k++ )
                    dp[u][j] = min ( dp[u][j] , dp[v][k]+dp[u][j-k]-1 );
        }
    }
    ans = min ( ans , dp[u][m]+1 );
}

int main()
{
    while( scanf ( "%d%d" , &n , &m )==2 )
    {
        tol = 0;
        ans = inf;
        memset ( head , -1 , sizeof(head) );
        memset ( dp , inf , sizeof(dp) );
        memset ( indeg , 0 , sizeof(indeg) );
        for ( int i=1 ; i<n ; i++ )
        {
            int u,v;
            scanf ( "%d%d" , &u , &v );
            addedge ( u , v );
            addedge ( v , u );
            indeg[u]++;
            indeg[v]++;
        }
        dfs( 1 , 0 );
        printf ( "%d\n" , ans );
    }
    return 0;
}