树形DP-Bribing FIPA

Bribing FIPA

 （树形DP）

题意：给定n，m，接下来n行每行给一个城市，给出到城市的费用和与通向它的城市（可能没有）。

问你至少访问m个城市的情况下，使费用最小。

思路：

就是一个树形DP节点的分配问题，状态转移方程为 dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[z][k]);//分配k个要访问的节点给z子节点

注意初始化，具体见代码。

最坑的是输入。用一个map把字符串映射为数，用gets读入判断‘#’，sscanf 取出s中的整数m和n，ch=getchar()判断输入费用之后知否还有别的城市。构造0为根。输入是从网上看的，长见识了。这道题还是有难度的，不过我觉得主要难在初始化和输入上。

#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<string>
#include<queue>
#include<vector>
#include<map>
using namespace std;
const int mx=205;
const int inf=0x3f3f3f3f;
struct node{
 int w,v,nex;
}a[mx<<1];
int t,dp[mx][mx],fa[mx],c[mx],h[mx],sum[mx],n,m;
map<string,int> mp;//city映射为数来建树
void add(int z,int f)
{
    a[t].v=z;
    a[t].nex=h[f];
    h[f]=t++;
}
void dfs1(int u)
{
 sum[u]=1;//统计这一个节点的数目 
 dp[u][0]=0;//访问0个点费用为0
 int j,k;
 for(int i=h[u];i!=-1;i=a[i].nex)
 {
  int z=a[i].v;
  dfs1(z);
  sum[u]+=sum[z];
  for(j=sum[u];j>=1;j--){
  for(k=0;k<=j&&k<=sum[z];k++) dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[z][k]);//访问子树的k个点
 }
 }

 dp[u][sum[u]]=c[u];//这是关键。遍历完整颗子树，则值为c[u]

}
int main()
{
    int T,ans,num;
    char s[mx],sz[mx],sf[mx],ch;
    while(gets(s)&&s[0]!='#')//gets 读入带空格的字符串
    {
     sscanf(s,"%d%d",&n,&m);//sscanf从s中取出整数n和m，新学的，记住
     memset(h,-1,sizeof(h));
     memset(fa,-1,sizeof(fa));
     memset(dp,0x3f,sizeof(dp));
     memset(sum,0,sizeof(sum));
     mp.clear();num=0;t=0;//注意初始化
     int w;
     for(int i=1;i<=n;i++)
     {
      scanf("%s%d",s,&w);
      if(mp.count(s)==0) mp[s]=++num;//建立映射
      c[mp[s]]=w;
      while(1)
      {
          ch=getchar();
          if(ch=='\n') break;//输入回车说明该组输入已经结束
          scanf("%s",sz);
          if(mp.count(sz)==0) mp[sz]=++num;
          add(mp[sz],mp[s]);
          fa[mp[sz]]=mp[s];//记录父节点，最后没有父节点的都连到0,0即为树的根
      }
     }
     for(int i=1;i<=n;i++) if(fa[i]==-1) add(i,0);//连接根
     sum[0]=n;
     dfs1(0);
     ans=inf;
     for(int i=m;i<=n;i++) ans=min(ans,dp[0][i]);//最小值即为能访问至少m个城市的最小费用
     printf("%d\n",ans);
    }
    return 0;
}