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