JAVA程序设计：每棵子树内缺失的最小基因值（LeetCode：2003）

有一棵根节点为 0 的 家族树 ，总共包含 n 个节点，节点编号为 0 到 n - 1 。给你一个下标从 0 开始的整数数组 parents ，其中 parents[i] 是节点 i 的父节点。由于节点 0 是 根 ，所以 parents[0] == -1 。

总共有 105 个基因值，每个基因值都用 闭区间 [1, 105] 中的一个整数表示。给你一个下标从 0 开始的整数数组 nums ，其中 nums[i] 是节点 i 的基因值，且基因值 互不相同 。

请你返回一个数组 ans ，长度为 n ，其中 ans[i] 是以节点 i 为根的子树内 缺失 的 最小 基因值。

节点 x 为根的 子树 包含节点 x 和它所有的 后代 节点。

示例 1：

输入：parents = [-1,0,0,2], nums = [1,2,3,4]
输出：[5,1,1,1]
解释：每个子树答案计算结果如下：
- 0：子树包含节点 [0,1,2,3] ，基因值分别为 [1,2,3,4] 。5 是缺失的最小基因值。
- 1：子树只包含节点 1 ，基因值为 2 。1 是缺失的最小基因值。
- 2：子树包含节点 [2,3] ，基因值分别为 [3,4] 。1 是缺失的最小基因值。
- 3：子树只包含节点 3 ，基因值为 4 。1是缺失的最小基因值。
示例 2：

输入：parents = [-1,0,1,0,3,3], nums = [5,4,6,2,1,3]
输出：[7,1,1,4,2,1]
解释：每个子树答案计算结果如下：
- 0：子树内包含节点 [0,1,2,3,4,5] ，基因值分别为 [5,4,6,2,1,3] 。7 是缺失的最小基因值。
- 1：子树内包含节点 [1,2] ，基因值分别为 [4,6] 。 1 是缺失的最小基因值。
- 2：子树内只包含节点 2 ，基因值为 6 。1 是缺失的最小基因值。
- 3：子树内包含节点 [3,4,5] ，基因值分别为 [2,1,3] 。4 是缺失的最小基因值。
- 4：子树内只包含节点 4 ，基因值为 1 。2 是缺失的最小基因值。
- 5：子树内只包含节点 5 ，基因值为 3 。1 是缺失的最小基因值。
示例 3：

输入：parents = [-1,2,3,0,2,4,1], nums = [2,3,4,5,6,7,8]
输出：[1,1,1,1,1,1,1]
解释：所有子树都缺失基因值 1 。
 

提示：

n == parents.length == nums.length
2 <= n <= 105
对于 i != 0 ，满足 0 <= parents[i] <= n - 1
parents[0] == -1
parents 表示一棵合法的树。
1 <= nums[i] <= 105
nums[i] 互不相同。

思路：在评论区看到一个不错的解题方法存一下【来自带佬】：

该方法主要由两个关键思路组成

• 剖出包含基因值为1的节点的链，非此链的节点答案都是1
• 这条链上的由深到浅的节点答案是非严格递增的

所以根据提示1，先找到基因值为1的节点，然后从深到浅一直处理到根节点。根据提示2，维护一个访问数组，处理到一个节点u时，答案从上一个节点的答案开始寻找，在此之前需要将u节点的其他子树基因值标记为已访问，那么寻找的第一个未访问的基因值就是u节点的缺失最小基因值，也是下一个较浅节点答案的搜索起点。 时间复杂度O(n)

class Solution {
    private int[] ans;
    private boolean[] used;
    private List<List<Integer>> e;

    public int[] smallestMissingValueSubtree(int[] parents, int[] nums) {
        e = new ArrayList<>();
        int n = parents.length;
        ans = new int[n];
        used = new boolean[n + 10];
        for (int i = 0; i < n; i++)
            e.add(new ArrayList<>());
        for (int i = 1; i < n; i++)
            e.get(parents[i]).add(i);
        Arrays.fill(ans, 1);
        for (int i = 0; i < n; i++) {
            if (nums[i] == 1) {
                int u = i, v = i, now = 1;
                Queue<Integer> q = new LinkedList<>();
                while (u != -1) {
                    for (int j = 0; j < e.get(u).size(); j++) {
                        int nxt = e.get(u).get(j);
                        if (nxt != v)
                            q.add(e.get(u).get(j));
                    }
                    used[nums[u]] = true;
                    while (!q.isEmpty()) {
                        int top = q.poll();
                        used[nums[top]] = true;
                        for (int j = 0; j < e.get(top).size(); j++)
                            q.add(e.get(top).get(j));
                    }
                    for (int j = now; j <= 100005; j++) {
                        if (used[j])
                            continue;
                        ans[u] = now = j;
                        break;
                    }
                    v = u;
                    u = parents[u];
                }
                break;
            }
        }
        return ans;
    }
}