题目大意:
一些节点之间有道路相连,所有的节点和道路形成了一棵树。一共有 n 个节点,这些节点被编号为 1~n 。人类将 k 个机器人送上了火星,目的是采集这些金属。这些机器人都被送到了一个指定的着落点, S 号节点。每个机器人在着落之后,必须沿着道路行走。当机器人到达一个节点时,它会采集这个节点蕴藏的所有金属矿。当机器人完成自己的任务之后,可以从任意一个节点返回地球。当然,回到地球的机器人就无法再到火星去了。我们已经提前测量出了每条道路的信息,包括它的两个端点 x 和 y,以及通过这条道路需要花费的能量 w 。我们想花费尽量少的能量采集所有节点的金属,这个任务就交给你了。
解题思路:
子孙结点之间构成了不同组的背包,而机器人的个数就是背包的容量,耗费的能量就是价值;
dp[i][j]表示从i结点出发j个机器人可以遍历所有子孙结点所需要的最小花费;
转移方程dp【i】【j】=min(dp[i][j],dp[i][j-k]+k*edge[i].w+dp[v][k]);
上面的k是从1开始的,保证子孙状态肯定有机器人,但是即使子孙结点不放机器人,我们也需要访问v,因此我们每次进行状态转移的时候要把每一种状态的都加上0的这种情况dp[i][j]+=dp[v][0]+2*edge[i].w;这种状态表示的去了又回来;
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 100010
int head[maxn];
int cnt;
bool vis[maxn];
int dp[maxn][11];
int n,s,sum;
struct Edge
{
int to;
int w;
int next;
}edge[3*maxn];
void init()
{
cnt=0;
memset(vis,0,sizeof(vis));
memset(head,-1,sizeof(head));
memset(dp,0,sizeof(dp));
}
void addedge(int u,int v,int c)
{
edge[cnt].to=v;
edge[cnt].w=c;
edge[cnt].next=head[u];
head[u]=cnt++;
}
void treedp(int s)
{
vis[s]=1;
for(int i=head[s];i!=-1;i=edge[i].next)
{
int v=edge[i].to;
if(vis[v])
continue;
treedp(v);
for(int j=sum;j>=0;j--)
{
dp[s][j]+=dp[v][0]+2*edge[i].w;
for(int k=1;k<=j;k++)
dp[s][j]=min(dp[s][j],dp[s][j-k]+k*edge[i].w+dp[v][k]);
}
}
}
int main()
{
while(~scanf("%d%d%d",&n,&s,&sum))
{
init();
for(int i=0;i<n-1;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
addedge(a,b,c);
addedge(b,a,c);
}
treedp(s);
printf("%d\n",dp[s][sum]);
}
}