poj 3345 Bribing FIPA(树形dp)

题意：

给出n个节点的一个森林，现在要选出至少m个节点，每个节点都有一个价值，如果选了树根对应的子树可以不用花费价值就可以选到，求最小的价值。

题解：

状态dp[i][j] i为根的树选j个节点的最小价值。

整个就一背包，注意初始化就好。

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<map>
using namespace std;
//typedef __int64 lld;
#define oo 0x3f3f3f3f
#define maxn 205
int dp[maxn][maxn];
struct EDGE
{
    int v,next;
}E[maxn<<1];
int num[maxn],val[maxn];
int head[maxn],tol;
int father[maxn];
map<string,int> mat;

void inst()
{
    memset(head,-1,sizeof head);
    tol=0;
    memset(dp,0x3f,sizeof dp);
    memset(num,0,sizeof num);
    memset(father,-1,sizeof father);
}

void add_edge(int u,int v)
{
    E[tol].v=v;
    E[tol].next=head[u];
    head[u]=tol++;
}


void tree_dp(int u)
{
    num[u]=1;
    dp[u][0]=0;
    for(int i=head[u];i!=-1;i=E[i].next)
    {
        int v=E[i].v;
        tree_dp(v);
        num[u]+=num[v];
        for(int j=num[u];j>=1;j--)
            for(int k=0;k<=j&&k<=num[v];k++)
                dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[v][k]);
    }
    dp[u][num[u]]=val[u];
}

int main()
{
    int u,v,T,m,n,ans,cnt,w;
    char s[200],ss[200],c;
    while(gets(s)&&s[0]!='#')
    {
        sscanf(s,"%d%d",&n,&m);
        inst();
        mat.clear();
        cnt=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%s%d",s,&w);
            if(!mat[s]) mat[s]=++cnt;
            val[mat[s]]=w;
            while(1)
            {
                c=getchar();
                if(c=='\n') break;
                scanf("%s",ss);
                if(!mat[ss]) mat[ss]=++cnt;
                add_edge(mat[s],mat[ss]);
                father[mat[ss]]=mat[s];
            }

        }
        num[0]=n;
        for(int i=1;i<=n;i++)
            if(father[i]==-1)
                add_edge(0,i);
        tree_dp(0);
        ans=oo;
        for(int i=m;i<=n;i++)
            ans=min(dp[0][i],ans);
        printf("%d\n",ans);
    }
    return 0;
}