bzoj4987 Tree 树形dp

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;
}