leetcode2003. 每棵子树内缺失的最小基因值

题目描述

有一棵根节点为 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 是缺失的最小基因值。

解题思路

引用:题解
在这里插入图片描述

class Solution {
public:
    vector<int> smallestMissingValueSubtree(vector<int>& parents, vector<int>& nums) {
        int n = parents.size();
        vector<int> ans(n, 1);
        // auto it = find(nums.begin(), nums.end(), 1);//寻找数组中是否有1
        // if (it == nums.end()) return ans;
        //建图
        vector<vector<int>> g(n);
        for (int i = 1; i < n; i ++ ) //必须从1开始,因为第一个为-1,g[-1]不合法
            g[parents[i]].push_back(i);

        int cur = -1;//记录基因值为1的坐标
        for (int i = 0; i < nums.size(); i ++ ) 
            if (nums[i] == 1) cur = i;
        if (cur == -1) return ans;//不存在基因值为1的节点,那么答案都是1

        unordered_set<int> vis;//记录当前已有的基因值
        function<void(int)> dfs = [&](int x) -> void {
            vis.insert(nums[x]);//将遍历到的基因值加入哈希表
            for (auto v: g[x]) 
                if (!vis.count(nums[v])) dfs(v);
        };

        int val = 2;
        while (cur >= 0) {
            dfs(cur);

            while (vis.count(val)) val ++ ;//当前基因值存在,基因值++ 
            ans[cur] = val;//更新答案
            cur = parents[cur];//当前点走到父节点
        }
        return ans;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值