传送门
题目的意思就是让找出
k
k
k个点的虚树然后按顺序遍历,除了
(
a
1
,
a
k
)
(a1,ak)
(a1,ak)上的边经过一次外其他边都是经过两次
所以最优解就是找出一个
k
k
k个点的子树,直径上的边只经过一次,其他边经过两次
设 f [ i ] [ j ] [ 0 / 1 / 2 ] f[i][j][0/1/2] f[i][j][0/1/2]表示 u u u为根的子树,选了 j j j个点,直径端点确定了 0 / 1 / 2 0/1/2 0/1/2个时的最优解,就像树形背包那样枚举转移,同样也要注意不能直接在 f f f数组上改
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 3005
#define inf 0x3f3f3f3f
using namespace std;
int n,m,cnt,head[N],f[N][N][3],siz[N],ff[N][3],ans;
inline int rd(){
int x=0,f=1;char c=' ';
while(c<'0' || c>'9') f=c=='-'?-1:1,c=getchar();
while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
return x*f;
}
struct EDGE{
int to,nxt,w;
}edge[N<<1];
inline void add(int x,int y,int z){
edge[++cnt].to=y; edge[cnt].nxt=head[x]; edge[cnt].w=z; head[x]=cnt;
}
inline void DP(int u,int fa){
siz[u]=1;
f[u][1][0]=f[u][1][1]=f[u][0][0]=0;
for(int i=head[u];i;i=edge[i].nxt){
int v=edge[i].to; if(v==fa) continue;
DP(v,u);
memcpy(ff,f[u],sizeof f[u]);
for(int j=0;j<=siz[u];j++)
for(int k=0;k<=siz[v];k++){
ff[j+k][0]=min(ff[j+k][0],f[u][j][0]+f[v][k][0]+edge[i].w*2);
ff[j+k][1]=min(ff[j+k][1],f[u][j][0]+f[v][k][1]+edge[i].w);
ff[j+k][1]=min(ff[j+k][1],f[u][j][1]+f[v][k][0]+edge[i].w*2);
ff[j+k][2]=min(ff[j+k][2],f[u][j][1]+f[v][k][1]+edge[i].w);
ff[j+k][2]=min(ff[j+k][2],f[u][j][2]+f[v][k][0]+edge[i].w*2);
ff[j+k][2]=min(ff[j+k][2],f[u][j][0]+f[v][k][2]+edge[i].w*2);
}
siz[u]+=siz[v];
for(int j=1;j<=siz[u];j++)
f[u][j][0]=ff[j][0],f[u][j][1]=ff[j][1],f[u][j][2]=ff[j][2];
}
ans=min(ans,f[u][m][2]);//
}
int main(){
n=rd(); m=rd();
for(int i=1;i<n;i++){
int x=rd(),y=rd(),z=rd();
add(x,y,z); add(y,x,z);
}
memset(f,0x3f,sizeof f); ans=inf;
DP(1,0);
printf("%d\n",ans);
return 0;
}