[ 树形DP ] BZOJ4987

显然最终路径不可能跳过一个没取过的点。
记 $3$ 个数组分别表示两端在根、一端在根、两端都不在根的有 $i$ 个点的路径长度最小值。
然后做一次树形背包即可。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=3010;
const int INF=1e9;
int k,n,m,p;
int f[N][N],g[N][N],h[N][N],sz[N];
int F[N],G[N],H[N];
int hh[N],nx[N<<1],t[N<<1],w[N<<1],num;
int x,y,z;
int Ans=INF;
inline void Add(int x,int y,int z){
    t[++num]=y;w[num]=z;nx[num]=hh[x];hh[x]=num;
}
void Dfs(int x,int y){
    f[x][1]=g[x][1]=h[x][1]=0;sz[x]=1;
    for(int i=hh[x];i;i=nx[i]){
        int v=t[i];
        if(v==y)continue;
        Dfs(v,x);
        for(int j=1;j<=sz[x];j++)F[j]=f[x][j],G[j]=g[x][j],H[j]=h[x][j];
        for(int j=sz[x]+1;j<=min(m,sz[x]+sz[v]);j++)F[j]=G[j]=H[j]=INF;
        for(int j=1;j<=sz[v];j++)
            for(int k=min(m-j,sz[x]);k>=1;k--){
                F[j+k]=min(F[k+j],min(f[v][j]+h[x][k]+w[i],h[v][j]+f[x][k]+(w[i]<<1)));
                G[j+k]=min(G[k+j],min(f[v][j]+f[x][k]+w[i],h[v][j]+g[x][k]+(w[i]<<1)));
                G[j+k]=min(G[k+j],g[v][j]+h[x][k]+(w[i]<<1));
                H[j+k]=min(H[k+j],h[v][j]+h[x][k]+(w[i]<<1));
            }
        sz[x]+=sz[v];
        for(int j=1;j<=sz[x];j++)f[x][j]=F[j],g[x][j]=G[j],h[x][j]=H[j];
    }
    Ans=min(Ans,min(f[x][m],g[x][m]));
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)scanf("%d%d%d",&x,&y,&z),Add(x,y,z),Add(y,x,z);
    memset(f,61,sizeof(f));memset(g,61,sizeof(g));memset(h,61,sizeof(h));
    Dfs(1,0);
    cout<<Ans<<endl;
    return 0;
}