二叉苹果树

题目传送门

思路

这道题就是一道比较经典的树形$DP$。

我们令$f[i][j]$为以$i$为根，保留$j$条边的最大值 。

我们的状态转移方程式就为:

f[u][j]=max(f[u][j],f[u][j-k-1]+f[v][k]+w);

我们需要保留$j$条树枝，我们就在以$u$为根的树中选择另外$j-k-1$条树枝，然后再在以$v$为根的树中再选择$k$条树枝。

那有些人就有疑问了，我们还有一条树枝呢？就是$u\to v$这一条。

---

代码

#include <bits/stdc++.h>
using namespace std;
int n,q;
int head[105],ver[205],edge[205],nxt[205],cnt;
int f[105][105],size[105];

void read(int &x) {
    int f=1; x=0; char ch=getchar();
    while(ch<'0' || ch>'9') { if(ch=='-') f=-1; ch=getchar(); }
    while(ch>='0' && ch<='9') { x=(x<<1)+(x<<3)+(ch^48); ch=getchar(); }
    x*=f;
}

void add_edge(int u,int v,int w) { ver[++cnt]=v,edge[cnt]=w,nxt[cnt]=head[u],head[u]=cnt; }

void dfs(int u,int fa) {
	for(int i=head[u];i;i=nxt[i]) {
		int v=ver[i],w=edge[i];
		if(v==fa) continue;
		dfs(v,u);
		size[u]+=size[v]+1; //记录一下我们以u的树和以v的树共有多少条树枝
		for(int j=min(size[u],q);j;j--) {
			for(int k=min(j-1,size[v]);k>=0;k--)
				f[u][j]=max(f[u][j],f[u][j-k-1]+f[v][k]+w);
		}
	}
	return ;
}

int main() {
	read(n),read(q);
	for(int i=1,u,v,w;i<n;i++) {
		scanf("%d %d %d",&u,&v,&w);
		add_edge(u,v,w),add_edge(v,u,w);
	}
	dfs(1,-1);
	printf("%d",f[1][q]);
	return 0;
}