洛谷 P2015 二叉苹果树

P2015 二叉苹果树

和选课差不多

但是并不是每条边的两边 谁是谁的父亲。。

根据边权确立点权

然后转二叉树

树形dp

#include<bits/stdc++.h>
using namespace std;
#define maxn 1999
int n,m,tot,head[maxn],fa[maxn],ls[maxn],rs[maxn],w[maxn],dp[maxn][maxn];
struct Edge{
    int to,dis,next;
}edge[maxn]; 

void add(int u,int v,int d)
{
    edge[++tot].to=v; edge[tot].dis=d; edge[tot].next=head[u]; head[u]=tot;
    edge[++tot].to=u; edge[tot].dis=d; edge[tot].next=head[v]; head[v]=tot;
}

void get(int now)
{
    for(int i=head[now];i;i=edge[i].next)
    {
        if(fa[now]!=edge[i].to)
        {
            fa[edge[i].to]=now;
            w[edge[i].to]+=edge[i].dis;
            get(edge[i].to);
        }
    }
}

int dfs(int i,int j)
{
    if(i>n||j>m+1||i<1||j<1) return 0;
    if(dp[i][j]) return dp[i][j];
    for(int k=0;k<j;k++)
        dp[i][j]=max(dp[i][j],dfs(ls[i],k)+dfs(rs[i],j-k-1)+w[i]);
    dp[i][j]=max(dp[i][j],dfs(rs[i],j));
    return dp[i][j];
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
    }
    add(1,0,0);
    get(0);
    for(int i=1;i<=n;i++)
    {
        if(!ls[fa[i]]) ls[fa[i]]=i;
        else{
            int dad=ls[fa[i]];
            while(rs[dad]) dad=rs[dad];
            rs[dad]=i;
        }
    }
    printf("%d
",dfs(ls[0],m+1));
    return 0;
}