java 牛客网之[动态规划中等]NC15 小红的树

最新推荐文章于 2025-03-12 09:34:45 发布

关切得大神

最新推荐文章于 2025-03-12 09:34:45 发布

阅读量4.4k

点赞数 1

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

本文链接：https://blog.csdn.net/qq_41115379/article/details/121202059

版权

LeetCode 专栏收录该内容

326 篇文章

订阅专栏

这篇博客介绍了如何使用树形动态规划（DP）解决一类有根树的染色问题。在给定的一棵有根树中，根节点编号为1，小红需要染一部分节点为红色，并对多次询问进行回答，每次询问指定节点子树内红色节点的数量。作者通过DFS遍历树结构，计算每个节点子树内的红色节点数，并存储在dp数组中，从而有效地回答查询。示例代码展示了如何实现这一算法。

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

题目的链接在这里：https://www.nowcoder.com/practice/66ab364d3fba487eb39bd3460fd484c0

题目大意

小红拿到了一棵有根树。根节点为1号节点。所谓树，指没有回路的无向连通图。现在小红想给一部分点染成红色。之后她有次询问，每次询问某点的子树红色节点的个数。

一、示意图

在这里插入图片描述

二、解题思路

树形dp

树形dp

代码如下：

import java.util.*;

public class  Main{
    /**
     * 一颗有根树 根节点为一号节点 把一部分染成红色 有q次询问 每次询问某点的子树红色节点的个数
     * 使用到了dp和dfs
     */
    static int[] dp;
    static String str;
    static ArrayList<ArrayList<Integer>> graph;
    public static void main(String[] args){
        Scanner sc=new Scanner(System.in);
        int n=sc.nextInt();
        //+10可能就是为了把范围扩大
        dp=new int[n+10];
        graph=new ArrayList<>(n+10);
        //然后开始初始化
        for(int i=0;i<=n;i++){
            graph.add(new ArrayList<>());
        }
        //然后是p n-1个整数 代表第2个节点到第n个节点的父亲
        for(int i=2;i<=n;i++){
            int p=sc.nextInt();
            //p就是对应节点的父亲 表示就是p这个位置的子节点是i
            graph.get(p).add(i);
        }

        //然后是 RW这些数值 next和nextLine的区别
        str=sc.next();
        //然后开始递归
        DFS(1,-1);
        //然后这就更新结束了
        int q=sc.nextInt();
        while (q-->0){
            //那就开始判断
            int x=sc.nextInt();
            System.out.println(dp[x]);
        }
    }

    private static int DFS(int x, int p) {
        //目的是求一个x子树的红色节点 那就是说dp[i]表示的就是 i的子树红色节点数量 p表示的是父节点？
        int sum=0;
        //先判断这个节点是不是红色的
        if(str.charAt(x-1)=='R'){
            sum++;
        }
        //然后再判断x的子节点
        for(int i=0;i<graph.get(x).size();i++){
            //就是找x的子节点数 有0-size-1个
            if(graph.get(x).get(i)!=p){
                //还会等于p吗
                sum+=DFS(graph.get(x).get(i),x);
            }
        }
        dp[x]=sum;
        return sum;
    }
}