社区讨论

全WA代码求调(改过必关)

P2015二叉苹果树参与者 2已保存回复 1

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
1 条
当前快照
1 份
快照标识符
@mirba534
此快照首次捕获于
2025/12/04 18:47
3 个月前
此快照最后确认于
2025/12/06 17:15
3 个月前
查看原帖
CPP
#include<bits/stdc++.h>
#define int long long
#define double long double
using namespace std;
const int N=1e2+10;
int n,m;
int a[N];
int dp[N][N];
int l[N],r[N];
vector<pair<int,int> >g[N];
void build(int u,int fa){
	for(int i=0;i<g[u].size();i++){
		int v=g[u][i].first,x=g[u][i].second;
		if(v==fa)continue;
		a[v]=x;
		if(!l[u])l[u]=v;
		else r[u]=v;
		build(v,u);
	}
}
int dfs(int u,int m){
	if(m==0)return 0;
	if(l[u]==0&&r[u]==0)return 0;
	if(dp[u][m])return dp[u][m];
	for(int k=0;k<=m-1;k++){
		int l_=dfs(l[u],k);
		if(k>0)l_+=a[l[u]];
		
		int r_=dfs(r[u],m-k-1);
		if(m-k-1>0)r_+=a[r[u]];
		
		dp[u][m]=max(dp[u][m],l_+r_);
	}
	return dp[u][m];
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	memset(dp,-1,sizeof(dp));
	cin>>n>>m;
	for(int i=1;i<n;i++){
		int u,v,x;
		cin>>u>>v>>x;
		g[u].push_back({v,x});
		g[v].push_back({u,x});
	}
	build(1,0);
	cout<<dfs(1,m);
	return 0;
}

回复

1 条回复,欢迎继续交流。

正在加载回复...