二叉苹果树
题目链接
思路:
对于x为根的子树最大值可以从左右儿子选,只需要遍历左右儿子保留的数量即可。但由于题目要求与根节点(1)相连,也就是说如果选了左儿子,左儿子与根x的边就要连。由于给的是边权,先把边权转化成子节点的点权。
代码:
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
#define fi first
#define se second
const ll mod=1e9+7;
const int N=2e5+10;
#define int ll
const double eps=1e-6;ll qpow(ll a,ll b){ll res=1;while(b){if(b&1) res=(res*a)%mod;a=(a*a)%mod;b>>=1;}return res;
}int n,q;
vector<int> v[3000],l[3000];
int dp[500][500];
int a[3000];void c(int x,int f,int value){a[x]=value;for(int i=0;i<v[x].size();i++){int j=v[x][i],w=l[x][i];if(j==f) continue;c(j,x,w);}
}void dfs(int x,int f){vector<int> p;for(int k=0;k<v[x].size();k++){int y=v[x][k];if(y==f) continue;dfs(y,x);/*for(int i=q;i>=1;i--){for(int j=0;j<i;j++){dp[x][i]=max(dp[x][i],dp[x][i-j-1]+dp[y][j]+a[y]);}}*/p.push_back(y);}if(p.size()==0) return;int t=p[0],tt=p[1];for(int i=0;i<=q;i++){for(int j=0;j<=q;j++){if(!i&&!j) continue;if(i&&!j){dp[x][i+j]=max(dp[x][i+j],dp[t][i-1]+a[t]);}else if(!i&&j){dp[x][i+j]=max(dp[x][i+j],dp[tt][j-1]+a[tt]);}else{dp[x][i+j]=max(dp[x][i+j],dp[t][i-1]+dp[tt][j-1]+a[t]+a[tt]);}}}
}void solve(){while(cin>>n>>q){for(int i=0;i<=n;i++){v[i].clear();}for(int i=1;i<n;i++){int x,y,z;cin>>x>>y>>z;v[x].push_back(y);v[y].push_back(x);l[x].push_back(z);l[y].push_back(z);}c(1,0,0);dfs(1,0);cout<<dp[1][q]<<endl;}}signed main(){ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);int t=1;//cin>>t;while (t--){solve();}return 0;
}