原题链接:https://www.luogu.org/problemnew/show/P2015
题目大意
给定有根二叉树,保留Q条边的前提下,边权和最大。
解法
直接暴力枚举两个子树的情况。
#include <iostream>
#include <ctime>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#define forn(i, n) for (int i = 0; i < (int)(n); ++i)
#define for1(i, n) for (int i = 1; i <= (int)(n); ++i)
#define pb push_back
using namespace std;
typedef long long LL;
const int N=100+5;
struct Edge{
int v,w;
Edge(int v1,int w1):v(v1),w(w1){}
};
vector<Edge> Adj[N];
int n,q,dp[N][N];
void dfs(int u,int fa){
if(Adj[u].size()==1&&fa==Adj[u][0].v) return;
vector<int> V,W;
forn(j,Adj[u].size()){
int v=Adj[u][j].v;
int w=Adj[u][j].w;
if(v!=fa){
V.pb(v);
W.pb(w);
dfs(v,u);
}
}
int lv=V[0];
int lw=W[0];
int rv=V[1];
int rw=W[1];
for(int i=0;i<=n-1;i++){
//dp[u][i]
if(i>0) {
dp[u][i]=max(dp[u][i],dp[rv][i-1]+rw);
dp[u][i]=max(dp[u][i],dp[lv][i-1]+lw);
}
for(int j=0;j<=i-2;j++){
dp[u][i]=max(dp[u][i],dp[lv][j]+dp[rv][i-2-j]+lw+rw);
}
}
}
int main() {
#ifdef LOCAL_DEFINE
freopen("P2015.in", "rt", stdin);
#endif
int u,v,w;
cin>>n>>q;
for1(i,n-1){
cin>>u>>v>>w;
Adj[u].pb(Edge(v,w));
Adj[v].pb(Edge(u,w));
}
dfs(1,0);
// forn(i,7){
// forn(j,7){
// cout<<dp[i][j]<<" ";
// }
// cout<<endl;
// }
cout<<dp[1][q]<<endl;
#ifdef LOCAL_DEFINE
cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
return 0;
}