蓝桥杯 算法提高 金属采集 （树形动态规划）

//  蓝桥杯 算法提高 金属采集 （树形动态规划）
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
const int N=100010;
const int M=12;
struct Node
{
    int child;
    int value;
}s;
int visit[N]={0};//判断是否遍历的节点
int dp[N][M]={0};
int cost[N]={0};
int n,dir,m;//n节点数,dir位置，m机器人数
vector<Node> no[N];
int Min(int a,int b)
{
    if(a<b)
    {
        return a;
    }
    return b;
}
void pr()
{
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=m;j++)
        {
            cout<<dp[i][j]<<" ";
        }
        cout<<endl;
    }
}
void DP(int root)//把子的状态都移到父的状态
{
    int i;
    for(i=0;i<=m;i++)
    {
        dp[root][i]=0;
    }
    visit[root]=1;
    for(i=0;i<no[root].size();i++)
    {
        int son=no[root][i].child;
        if(visit[son]) continue;
        DP(son);
        cost[son]=no[root][i].value;
        for(int mi=m;mi>=0;mi--)//节点拥有的火星数
        {
            dp[root][mi]+=dp[son][0]+cost[son]*2;
            for(int mj=1;mj<=mi;mj++)//进行动态的转移
            {
                dp[root][mi]=Min(dp[root][mi],dp[root][mi-mj]+mj*cost[son]+dp[son][mj]);
            }
        }
    }
}
int main()
{
    scanf("%d%d%d",&n,&dir,&m);
    int a1,a2,dis;
    while(scanf("%d%d%d",&a1,&a2,&dis)==3)
    {
        s.child=a1;
        s.value=dis;
        no[a2].push_back(s);
        s.child=a2;
        s.value=dis;
        no[a1].push_back(s);
    }
    DP(dir);
    printf("%d\n",dp[dir][m]);
    return 0;
}