树形动态规划

听微雨

已于 2023-03-20 20:11:10 修改

阅读量52

点赞数

分类专栏：算法学习文章标签：动态规划 java 算法

于 2023-03-20 19:06:52 首次发布

本文链接：https://blog.csdn.net/m0_54136420/article/details/129673825

版权

算法学习专栏收录该内容

21 篇文章 0 订阅

订阅专栏

该程序使用Java实现，基于状态转移方程处理一个包含6010个节点的树形结构。每个节点有两个状态（0和1），初始时计算每个节点的幸福值，然后通过深度优先搜索（DFS）遍历树，更新每个节点的状态值。最终找到根节点并输出其最大幸福值。

摘要由CSDN通过智能技术生成

没有上司的舞会

在这里插入图片描述

状态转移方程：
$\begin{aligned} &f(u,0)+=\sum max(f(s_i,0),f(s_i,1)) \\ &f(u,1)+=\sum f(s_i,0) \end{aligned}$

import java.io.*;
import java.util.*;

public class Main{
    static int N=6010;
    static int[] happy=new int[N];
    static int[] h=new int[N];
    static int[] e=new int[N];
    static int[] ne=new int[N];
    static int idx=0;
    static int[][] f=new int[N][2];
    static boolean[] has_father=new boolean[N];
    public static void main(String[] args)throws IOException{
        BufferedReader in=new BufferedReader(new InputStreamReader(System.in));
        int n=Integer.parseInt(in.readLine());
        for(int i=1;i<=n;i++)
            happy[i]=Integer.parseInt(in.readLine());
        String[] strs;
        Arrays.fill(h,-1);
        for(int i=0;i<n-1;i++){
            strs=in.readLine().split(" ");
            int a=Integer.parseInt(strs[0]);
            int b=Integer.parseInt(strs[1]);
            has_father[a]=true;
            add(b,a);
        }
        int root=1;
        while(has_father[root])root++;
        dfs(root);
        System.out.println(Math.max(f[root][0],f[root][1]));
    }
    static void add(int a,int b){
        e[idx]=b;ne[idx]=h[a];h[a]=idx++;
    }
    static void dfs(int u){
        f[u][1]+=happy[u];
        for(int i=h[u];i!=-1;i=ne[i]){
            int j=e[i];
            dfs(j);
            f[u][0]+=Math.max(f[j][0],f[j][1]);
            f[u][1]+=f[j][0];
        }
    }
}