Description
从前有棵树。
找出K个点A1,A2,…,Ak。
使得∑dis(AiAi+1),(1<=i<=K-1)最小。
I<=k<=n。
l<x,y<=n
1<=z<=10^5
n <= 3000
Solution
一开始还以为是dfs序上的dp
不难发现选出的点一定都是相邻的,为了和最小实际上就是选一条最长链的系数为1,其余系数为2
然后我们可以在树上dp。设f[i,j,k]表示以i为根的子树内选了j个点,最长链的端点选了k个的答案,转移的时候讨论一下当前边的贡献就可以了
根据某些奇奇怪怪的结论可以知道我们这样做是O(n^2)的
Code
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))
const int INF=0x3f3f3f3f;
const int N=3005;
struct edge {int y,w,next;} e[N*2];
int f[N][N][3],g[N][3];
int size[N],ans=INF,n,m;
int ls[N],edCnt;
int read() {
int x=0,v=1; char ch=getchar();
for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
return x*v;
}
void add_edge(int x,int y,int w) {
e[++edCnt]=(edge) {y,w,ls[x]}; ls[x]=edCnt;
e[++edCnt]=(edge) {x,w,ls[y]}; ls[y]=edCnt;
}
void dfs(int now,int fa) {
size[now]=1;
f[now][1][0]=f[now][1][1]=f[now][1][2]=0;
for (int i=ls[now];i;i=e[i].next) {
if (e[i].y==fa) continue;
dfs(e[i].y,now);
rep(j,0,size[now]+size[e[i].y]) g[j][0]=g[j][1]=g[j][2]=INF;
rep(j,0,size[now]) {
rep(k,0,size[e[i].y]) {
g[j+k][0]=std:: min(g[j+k][0],f[now][j][0]+f[e[i].y][k][0]+e[i].w*2);
g[j+k][1]=std:: min(g[j+k][1],f[now][j][1]+f[e[i].y][k][0]+e[i].w*2);
g[j+k][1]=std:: min(g[j+k][1],f[now][j][0]+f[e[i].y][k][1]+e[i].w);
g[j+k][2]=std:: min(g[j+k][2],f[now][j][0]+f[e[i].y][k][2]+e[i].w*2);
g[j+k][2]=std:: min(g[j+k][2],f[now][j][2]+f[e[i].y][k][0]+e[i].w*2);
g[j+k][2]=std:: min(g[j+k][2],f[now][j][1]+f[e[i].y][k][1]+e[i].w);
}
}
size[now]+=size[e[i].y];
rep(j,0,size[now]) rep(k,0,2) {
f[now][j][k]=std:: min(f[now][j][k],g[j][k]);
}
}
ans=std:: min(ans,f[now][m][2]);
}
int main(void) {
fill(f,63);
n=read(),m=read();
rep(i,2,n) {
int x=read(),y=read(),w=read();
add_edge(x,y,w);
}
dfs(1,0);
printf("%d\n", ans);
return 0;
}