比赛 动规 评测结果 AAAAAAAAAAAAA
题目名称 二叉苹果树 最终得分 100
用户昵称 Hyoi_0Koto 运行时间 0.000 s
代码语言 C++ 内存使用 0.00 MiB
提交时间 2017-06-18 21:34:42
显示代码纯文本
#include<cstdio>
#include<cctype>
#include<iostream>
using namespace std;
const int maxn=103;
int n,q,u,v,wi;
int f[maxn][maxn],g[maxn][maxn],w[maxn],g2[maxn][3];
inline void in(int &x){
    x=0;int f=1;char t=getchar();
    while(!isdigit(t)){if(t=='-')f=-1;t=getchar();}
    while(isdigit(t)){x=x*10+t-48;t=getchar();}
    x*=f;
}
inline void maketree(int v);
inline void buildtree(int x,int y,int s){
	w[y]=g[x][y];
	g2[x][s]=y;
	g[x][y]=-1;g[y][x]=-1;
	maketree(y);
}
inline void maketree(int v){
	int s=0;
	for(int i=0;i<=n;i++){
		if(g[v][i]>=0){
			s++;
			buildtree(v,i,s);
			if(s==2) return;
		}
	}
}
inline void dfs_dp(int v,int l){
	if(!l) f[v][l]=0;
	else{
		if((!g2[v][1])&&(!g2[v][2])) f[v][l]=w[v];
		else{
			for(int i=0;i<l;i++){
				if(f[g2[v][1]][i]<0) dfs_dp(g2[v][1],i);
				if(f[g2[v][2]][l-i-1]<0) dfs_dp(g2[v][2],l-i-1);
				f[v][l]=max(f[v][l],f[g2[v][1]][i]+f[g2[v][2]][l-i-1]+w[v]);
			}
		}
	}
}
inline int mian(){
	freopen("ecappletree.in","r",stdin);
	freopen("ecappletree.out","w",stdout);
	in(n);in(q);
	for(int i=0;i<=n;i++)
	 for(int j=0;j<=n;j++){
	 	g[i][j]=f[i][j]=-1;
	 }
	for(int i=1;i<n;i++){
		in(u);in(v);in(wi);
		g[u][v]=wi;g[v][u]=wi;
	}
	maketree(1);
	dfs_dp(1,q+1);
	printf("%d",f[1][q+1]);
	return 0;
}
int miku=mian();
int main(){;}