您的位置:首页 > 其它

hdu4003

2016-03-11 09:17 239 查看
链接:点击打开链接

题意:一棵N个节点的树,每条边都有一个权值,K个机器人从S点出发,问让机器人遍历所有边,最少花费值是多少

代码:

#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
int N,S,K;
int dp[10005][15],used[10005];                  //dp[i][j]代表以i为根节点的子树花费
struct node{                                    //j个机器人遍历的最少花费
int to,cost;
};
vector<node> G[10005];
void dfs(int s){
int i,j,k,tmp;
used[s]=1;
for(i=0;i<G[s].size();i++){
tmp=G[s][i].to;
if(G[tmp].size()&&!used[tmp]){
dfs(tmp);
for(j=K;j>=0;j--){                  //将一个机器人遍历以s为节点的树作为初始值
dp[s][j]+=(dp[tmp][0]+2*G[s][i].cost);
for(k=1;k<=j;k++)
dp[s][j]=min(dp[s][j],dp[s][j-k]+dp[tmp][k]+k*G[s][i].cost);
}                                   //树形dp
}
}
}
int main(){
int i,j,a,b,w;
while(scanf("%d%d%d",&N,&S,&K)!=EOF){
for(i=1;i<=N;i++)
G[i].clear();
memset(dp,0,sizeof(dp));
memset(used,0,sizeof(used));
for(i=0;i<N-1;i++){
scanf("%d%d%d",&a,&b,&w);
G[a].push_back((node){b,w});
G[b].push_back((node){a,w});
}
dfs(S);
printf("%d\n",dp[S][K]);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: