社区讨论
全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 条回复,欢迎继续交流。
正在加载回复...