Leetcode2003-每棵子树内缺失的最小基因值

16 篇文章 0 订阅

DFS

两次dfs,第一次dfs我们需要对于每个结点求一些额外的信息。考虑一个结点向下递归求其子结点的答案时,我们要传下去哪些信息呢,我们需要告诉子树它所缺失的最小值。这时又引入了一个问题,子树所缺失的最小值怎么求呢,分三部分:

①结点上方其他结点的最小值。

②本结点的值。

③以及除了这颗子树之外,其他子树中所有结点的最小值。

对这三部分取最小值,就是要传给子树的信息,即它所缺失的最小值。因此我们在第一遍dfs的时候需要记录的信息为某颗子树son1结点中的最小值及另外某颗子树son2结点中的次小值。之所以要记录次小值,是因为递归son1时,这时它是拥有最小值的,所以我们不能把最小值传下去,需要在其他子树中找到次小值传给他。为了直到最小值属于哪一颗子树,我们还需要开一个son数组来记录每个结点子树中的最小值在哪一颗子树上。

然后我们需要预处理出整个树缺失的最小值,作为递归开始的参数传入。因为所有值都是不同的,而结点只有n个,因此我们用一个appear数组来记录出现过的小于等于n的数字。最后找到前n+1个整数中没出现过的最小数即可。

第二次dfs就是直接得出答案了。

class Solution {
public:
    vector<int> smallestMissingValueSubtree(vector<int>& parents, vector<int>& nums) {
        int n=parents.size();
        vector<int> ans(n);
        vector<int> appear(n+1,0);
        vector<vector<int>> g(n);
        vector<vector<int>> v(n,vector<int>(2,-1));
        vector<int> son(n,-1);
        for(int i=0;i<n;i++){
            if(parents[i]!=-1){
                g[i].push_back(parents[i]);
                g[parents[i]].push_back(i);
            }
        }
        g[0].push_back(-1);
        for(int i=0;i<n;i++) if(nums[i]<=n) appear[nums[i]]=1;
        int k=n+1;
        for(int i=n;i>0;i--) if(!appear[i]) k=i;
        function<int(int,int)> dfs1=[&](int x,int fa){
            if(g[x].size()==1){
                v[x][0]=nums[x];
                return nums[x];
            }
            for(auto y:g[x]){
                if(y==fa) continue;
                int t=dfs1(y,x);
                if(v[x][0]==-1||t<v[x][0]){
                    v[x][1]=v[x][0];
                    v[x][0]=t;
                    son[x]=y;
                }else if(v[x][1]==-1||t<v[x][1]) v[x][1]=t;
            }
            return min(v[x][0],nums[x]);
        };
        dfs1(0,-1);
        function<void(int,int,int)> dfs2=[&](int x,int fa,int last){
            ans[x]=last;
            if(g[x].size()==1) return;
            for(auto y:g[x]){
                if(y==fa) continue;
                if(son[x]==y){
                    if(v[x][1]==-1) dfs2(y,x,min(last,nums[x]));
                    else dfs2(y,x,min({last,nums[x],v[x][1]}));
                }else dfs2(y,x,min({last,nums[x],v[x][0]}));
            }
        };
        dfs2(0,-1,k);
        return ans;
    }
};

时间复杂度:O(n)。

空间复杂度:O(n)。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值