大连网络赛 即 hdu 4003 Find Metal Mineral
这个题目我觉得应该可以做的,但是由于当时没看这个题,结果没做,现在看看,如果有时间,而且自己的状态非常好的话,也许是可以做出来的,
下面的代码是从网上掏的,自己最近没状态,懒得敲,先放在这儿供大家参考,
/*
题意:一棵有权树,从根结点中放入K个机器人,求用这K个机器人遍历所有的结点最少的权值和
分析:dp[i][j]表示对于以i结点为根结点的子树,放j个机器人所需要的权值和。
当j=0时表示放了一个机器人下去,遍历完结点后又回到i结点了。状态转移方程类似背包
*/
#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
const __int64 maxn=11000;
const __int64 maxm=21000;
struct edge
{
__int64 u,v,w,next;
}e[maxm];
__int64 edgeNum,first[maxn],dp[maxn][15];
void Addedge(__int64 u,__int64 v,__int64 w)
{
e[edgeNum].u=u,e[edgeNum].v=v,e[edgeNum].w=w,e[edgeNum].next=first[u],first[u]=edgeNum++;
e[edgeNum].u=v,e[edgeNum].v=u,e[edgeNum].w=w,e[edgeNum].next=first[v],first[v]=edgeNum++;
}
void DFS(__int64 t,__int64 p)
{
__int64 ii,i,j,k;
for(k=first[t];k!=-1;k=e[k].next)
{
i=e[k].v;
if(i==p) continue;
DFS(i,t);
}
//first[t]=-1时说明树中只有一个结点
if(first[t]==-1||(e[first[t]].next==-1&&p!=-1))//叶子结点,注意一定要把根结点排除。
{
for(i=0;i<=10;i++)
dp[t][i]=0;
return;
}
for(k=first[t];k!=-1;k=e[k].next)
{
i=e[k].v;
if(i==p) continue;
for(j=10;j>=0;j--)//跟01背包类似,方向不能换
{
if(dp[t][j]==-1)//第一棵子树
{
if(j)
{//最多放j个在该子树中,不一定要全部到该子树中(有一些机器人就停在这个结点上)。当然一个不放肯定不是最好的
for(ii=1;ii<=j;ii++)
if(dp[t][j]==-1||dp[t][j]>dp[i][ii]+ii*e[k].w)
dp[t][j]=dp[i][ii]+ii*e[k].w;
}
else dp[t][0]=dp[i][0]+2*e[k].w;
continue;
}
else//否则,若该子树放0个机器人
dp[t][j]+=dp[i][0]+2*e[k].w;
for(ii=1;ii<=j;ii++)
{
if(dp[t][j]>dp[t][j-ii]+dp[i][ii]+ii*e[k].w)
dp[t][j]=dp[t][j-ii]+dp[i][ii]+ii*e[k].w;
}
}
}
}
int main()
{
__int64 n,s,k,i,u,v,w;
while(scanf("%I64d%I64d%I64d",&n,&s,&k)!=EOF)
{
memset(first,-1,sizeof(first));
for(edgeNum=0,i=1;i<n;i++)
{
scanf("%I64d%I64d%I64d",&u,&v,&w);
Addedge(u,v,w);
}
memset(dp,-1,sizeof(dp));
DFS(s,-1);
printf("%I64d\n",dp[s][k]);
}
return 0;
}