【洛谷luogu】P2015-二叉苹果树(树形DP)

原题链接:https://www.luogu.org/problemnew/show/P2015

题目大意

给定有根二叉树,保留Q条边的前提下,边权和最大。

解法

直接暴力枚举两个子树的情况。

#include 
#include 
#include 
#include 
#include 
#include 
#include 

#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<
//		}
//		cout<
//	}
    cout<<dp[1][q]<<endl;



#ifdef LOCAL_DEFINE
    cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
    return 0;
}

你可能感兴趣的:(算法与数据结构)