您的位置:首页 > 其它

【树形dp】PKU-3345-Bribing FIPA

2012-03-26 20:54 302 查看
dp[i][j]表示在以i为根的树选j个国家的最少费用,具体思维看代码……

题目



#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#include<stack>
using namespace std;
#define FRE freopen("a.txt","r",stdin);
#define inf 999999999
#define N 205
vector<int> v
;
int m,w
,dp

;
int dfs(int now)
{
dp[now][0]=0;
int sum=1,t
;
int len=v[now].size();
for(int i=0;i<len;i++)
{
int u=v[now][i];
t[i]=dfs(u);
sum+=t[i];
}
if(now==0)sum--;
for(int i=0;i<len;i++)
{
int u=v[now][i];
for(int j=sum-1;j>=0;j--)              //逆序求,防止覆盖
{
if(dp[now][j]==inf)continue;
for(int k=1;k<=t[i];k++)
{
dp[now][j+k]=min(dp[now][j+k],dp[now][j]+dp[u][k]);
}
}
}
dp[now][sum]=w[now];              //把这个子树都选的费用
return sum;
}
int main()
{
//FRE;
int n,vis
;
while(scanf("%d%d",&n,&m)&&n)
{
for(int i=0;i<=n;i++)v[i].clear();
map<string,int> map;
map.clear();
int l=1;
memset(vis,0,sizeof(vis));
for(int i=0;i<n;i++)
{
char s[105];
scanf("%s",s);
if(!map[s])map[s]=l++;
int x=map[s];
scanf("%d",w+x);
while(getchar()==' ')              //这里要特别注意了,因为后面可能跟着很多国家
{
scanf("%s",s);
if(!map[s])map[s]=l++;
int y=map[s];
vis[y]=1;
v[x].push_back(y);
}
}
w[0]=0;              //这里忘了初始化,wa了几次
for(int i=1;i<=n;i++)
{
if(vis[i])continue;
v[0].push_back(i);              //增加一个虚拟根
w[0]+=w[i];
}
for(int i=0;i<=n;i++)
{
for(int j=0;j<=n;j++)
{
dp[i][j]=inf;
}
}
dfs(0);
int ans=inf;
for(int i=m;i<=n;i++)ans=min(ans,dp[0][i]);
printf("%d\n",ans);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: