题目链接
dp[i][j]表示以i为根的子树放j个机器人所产生的最小权值和,dp[i][0]表示放了一个机器人遍历i的子树然后返回i所产生的权值和。采用邻接表
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e5+10;
#define INF 0x3f3f3f3f
struct node{
int next,v,w;
}edge[N];
int n,s,k;//金属矿物,着陆点,机器人数量
int tot,head[N];
int dp[N][15];//dp[i][j]表示以 i为根的子树,放j个机器人所需要的权值和
void addadge(int u,int v,int w)
{
edge[tot].v=v;
edge[tot].w=w;
edge[tot].next=head[u];
head[u]=tot++;
}
void dfs(int u,int pre)
{
for(int i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
if(v==pre)continue;
dfs(v,u);
for(int j=k;j>=0;j--)
{
dp[u][j]+=dp[v][0]+2*edge[i].w;
//dp[i][0]表示放1个机器人,遍历节点后回到i节点
for(int l=1;l<=j;l++)
dp[u][j]=min(dp[u][j],dp[u][j-l]+dp[v][l]+l*edge[i].w);
}
}
}
int main()
{
//freopen("in.txt","r",stdin);
int i,u,v,w;
while(~scanf("%d%d%d",&n,&s,&k))
{
tot=0;
memset(head,-1,sizeof(head));
for(i=1;i<n;i++)
{
scanf("%d%d%d",&u,&v,&w);
addadge(u,v,w);
addadge(v,u,w);
}
memset(dp,0,sizeof(dp));
dfs(s,0);
printf("%d\n",dp[s][k]);
}
return 0;
}