POJ 2486 Apple Tree (树形dp 经典题)

EM-LGH

于 2018-08-24 19:18:09 发布

阅读量83

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/liyong1009s/article/details/82024239

本文深入探讨了一种基于图论的复杂算法，详细介绍了其数据结构、动态规划的应用以及递归函数的设计。通过实例展示了如何在图中寻找最优路径，特别关注了节点间的连接和权重的计算，为解决实际问题提供了有效的算法思路。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=105;
int head[maxn<<1],to[maxn<<1],val[maxn],nex[maxn<<1];
int F[maxn][maxn<<1],G[maxn][maxn<<1];
int cnt,V;
void addedge(int u,int v){
    nex[++cnt]=head[u],head[u]=cnt,to[cnt]=v;
}
void init(){
    memset(F,0,sizeof(F));
    memset(G,0,sizeof(G));
    memset(head,0,sizeof(head));
    memset(nex,0,sizeof(nex));
    cnt=0;
}
void dfs(int u,int fa){
    //for(int i=0;i<=V;++i)F[u][i]=G[u][i]=val[u];
    G[u][0]=F[u][0]=val[u];
    for(int v=head[u];v;v=nex[v])if(to[v]!=fa){
            dfs(to[v],u);
            for(int j=V;j>=0;--j)
                for(int k=0;k<=j;++k){
                    if(k>=2)G[u][j]=max(G[u][j],G[u][j-k]+G[to[v]][k-2]);
                    if(k>=2)F[u][j]=max(F[u][j],F[u][j-k]+G[to[v]][k-2]);
                    F[u][j]=max(F[u][j],G[u][j-k]+F[to[v]][k-1]);
                }
        }
}
int main(){
    int n;
    while(scanf("%d%d",&n,&V)!=EOF){
        init();
        for(int i=1;i<=n;++i)scanf("%d",&val[i]);
        for(int i=1;i<n;++i){
            int a,b;
            scanf("%d%d",&a,&b);
            addedge(a,b);
            addedge(b,a);
        }
        dfs(1,-1);
        int ans=0;
        for(int i=0;i<=V;++i)ans=max(ans,F[1][i]);
        printf("%d\n",ans);
    }
    return 0;
}