美团笔试附加题java题解2022.3.12

现在给你一颗树，每个树上的节点会被涂成黑色或白色。
现在定义好节点：
——对于白色的节点：若该节点没有子节点，或该节点子节点中至少有一个为黑色节点，则该节点是好节点
——对于黑色的节点：若该节点没有子节点，或该节点的所有子节点均为白色节点，则该节点是好节点

你的任务是找出这棵树上黑色的好节点和白色的好节点各有几个。

输入描述
第一行一个正整数n，表示这棵树共有n个节点，编号1到n。
第二行n个空格隔开的正整数，代表每个节点的颜色。0是白色，1是黑色。
接下来n个空格隔开的正整数，第 i 个正整数v表示节点 i 的父节点是v。其中，数字0表示这个节点是根。
1 <= n <= 10000

输出描述
一行两个正整数，以空格分开，第一个正整数表示白色的好节点有几个，第二个正整数表示黑色的好节点有几个。
样例输入
6
1 0 1 1 0 0
0 1 2 1 4 4
样例输出
3 2

**

主要思路为多叉树构建，然后dfs

**

import java.util.*;
public class Main{

    static int[] nums1 = new int[]{1,0, 1 ,0 ,1, 0};
    static int[] nums2 = new int[]{0 ,1, 2, 1, 4, 4};
    static int black = 0,white = 0;
    public static void main(String[] args) {
        Map<Integer, List<Integer>> root = new HashMap<>();
        for (int i = 0; i < nums2.length; i++) {
            root.put(i, new ArrayList<>());
        }
        for (int i = 1; i < nums2.length; i++) {
            int fatherIndex = nums2[i] - 1;
            List<Integer> value = root.get(fatherIndex);
            value.add(i);
            root.put(fatherIndex, value);
        }
        dfs(0, root);
        System.out.println(black + "," + white);
    }
    public static void dfs(int root, Map<Integer, List<Integer>> temp) {
        //没有子节点
        if(temp.get(root).size() == 0) {
            if (nums1[root] == 0) {
                white++;
            } else {
                black++;
            }
            return;
        }
        int sonBlack = 0;
        //不是叶子结点，挨个深度遍历到其子节点
        for (Integer node : temp.get(root)) {
                dfs(node,temp);
                //统计子节点黑色数量
                if (nums1[node] == 1) {
                    sonBlack++;
                }
        }
        if (nums1[root] == 1 && sonBlack == 0 ) black++;
        if (nums1[root] == 0 && sonBlack >= 1) white++;
    }
}