poj3345 树型DP

这题不难，输入较繁，典型的树形dp

题目大意：一共有n（<=200）个国家, 你想要用收买m(<=m)个国家，其中收买国家i的代价是v[ i ]。但是有些国家有拥属关系，如果A拥属B国，就是买通了A也意味着买通了B，而且这些关系是传递的。求最少需要付出的代价。

在树上背包，代码：

for (k=0;k<g[x].size();k++)
    {
        y=g[x][k];
        for (i=num[x];i>=0;i--)
            for (j=0;j<=i && j<=num[y];j++)
                if (dp[x][i-j]+dp[y][j]<dp[x][i])
                   dp[x][i]=dp[x][i-j]+dp[y][j];
    }

x为根，y是儿子，dp[x][i]表示以x为根的子树取i个的最小代价。（最好先把儿子都遍历完后再dp，这样比较清楚）

挺典型的，可以强化记忆。

这题的输入倒是有点e。。。

我用了sscanf，和sstream来搞。（表示c++是比pascal高级的多啊。。。）

还开了map。（第一次用map，挺方便的，但听说这东西巨慢。。。）

#include <cstdio>
#include <cstring>
#include <string>
#include <vector>
#include <map>
using namespace std;

const int maxn=205,oo=99999999;
vector<int> g[maxn];
int dp[maxn][maxn],v[maxn],num[maxn];
bool fa[maxn];
int n,m,id;

int dfs(int x)
{
    num[x]=1;
    int i,j,k,y;
    for (k=0;k<g[x].size();k++)
    {
        y=g[x][k];
        num[x]+=dfs(y);
    }
    for (k=0;k<=n;k++) dp[x][k]=oo;
    dp[x][0]=0;
    dp[x][num[x]]=v[x];
    for (k=0;k<g[x].size();k++)
    {
        y=g[x][k];
        for (i=num[x];i>=0;i--)
            for (j=0;j<=i && j<=num[y];j++)
                if (dp[x][i-j]+dp[y][j]<dp[x][i])
                   dp[x][i]=dp[x][i-j]+dp[y][j];
    }
    return num[x];
}
int main()
{
    freopen("pin.txt","r",stdin);
    freopen("pou.txt","w",stdout);
    char str[1000];
    int i,j,ans,now;
    while (gets(str))
    {
          if (str[0]=='#') break;
          sscanf(str,"%d%d",&n,&m);
          map<string,int> wmap;
          for (i=0;i<=n;i++)
              g[i].clear();
          memset(fa,0,sizeof(fa));
          id=0;
          for (i=1;i<=n;i++)
          {
              scanf("%s",str);
              if (wmap.find(str)==wmap.end()) 
                 wmap[str]=++id;
              now=wmap[str];
              scanf("%d",&v[now]);
              while (getchar()!='\n')
              {
                    scanf("%s",str);
                    if (wmap.find(str)==wmap.end())
                       wmap[str]=++id;
                    g[now].push_back(wmap[str]);
                    fa[wmap[str]]=true;
              }
          }
          v[0]=oo;
          for (i=1;i<=n;i++)
          {
              if (fa[i]) continue;
              g[0].push_back(i);
          }
          dfs(0);
          ans=oo;
          for (i=m;i<=n;i++)
              if (dp[0][i]<ans) ans=dp[0][i];
          printf("%d\n",ans);
    }
    return 0;
}