二叉苹果树（树形DP）

题面

原题

Solution

显然我们考虑这道题目可以很蠢的写一个Dp对吧。
考虑\(dp_i,j\)表示以i为根，保留j个节点的最大苹果数。
然后就可以树形Dp的写了。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<iostream>
#include<queue>
#define ll long long
#define file(a) freopen(a".in","r",stdin)//;freopen(a".out","w",stdout)
using namespace std;
inline int gi(){
	int sum=0,f=1;char ch=getchar();
	while(ch>'9' || ch<'0'){if(ch=='-')f=-f;ch=getchar();}
	while(ch>='0' && ch<='9'){sum=(sum<<3)+(sum<<1)+ch-'0';ch=getchar();}
	return f*sum;
}
inline ll gl(){
	ll sum=0,f=1;char ch=getchar();
	while(ch>'9' || ch<'0'){if(ch=='-')f=-f;ch=getchar();}
	while(ch>='0' && ch<='9'){sum=(sum<<3)+(sum<<1)+ch-'0';ch=getchar();}
	return f*sum;
}
const int maxn=110,maxq=110;
int dp[maxn][maxq],to[maxn<<1],nxt[maxn<<1],front[maxn],cnt,val[maxn],ans,son[maxn],fa[maxn],n,Q,w[maxn<<1],Son[maxn][3];
void Add(int u,int v,int W){
	to[++cnt]=v;w[cnt]=W;nxt[cnt]=front[u];front[u]=cnt;
}
void dfs(int u,int f){
	fa[u]=f;
	for(int i=front[u];i;i=nxt[i]){
		int v=to[i];
		if(v==f)continue;
		dfs(v,u);val[v]=w[i];
		son[u]++;Son[u][son[u]]=v;
	}
}
int tree_dp(int u,int q){
	if(!q)return dp[u][q]=0;
	if(!son[u])return val[u];
	if(dp[u][q]>-1)return dp[u][q];
	for(int j=0;j<q;j++)
		dp[u][q]=max(dp[u][q],tree_dp(Son[u][1],j)+tree_dp(Son[u][2],q-j-1)+val[u]);
	return dp[u][q];
}
int main(){
#ifndef ONLINE_JUDGE
	file("example");
#endif
	int i,j,m,k;
	memset(dp,-1,sizeof(dp));
	n=gi();Q=gi();Q++;
	for(i=1;i<n;i++){
		int u=gi(),v=gi(),W=gi();
		Add(u,v,W);Add(v,u,W);
	}
	dfs(1,1);
	printf("%d\n",tree_dp(1,Q));
	return 0;
}