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

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）。