题目描述:
你有一棵二叉树,这棵二叉树有个小问题,其中有且只有一个无效节点,它的右子节点错误地指向了与其在同一层且在其右侧的一个其他节点。
给定一棵这样的问题二叉树的根节点 root ,将该无效节点及其所有子节点移除(除被错误指向的节点外),然后返回新二叉树的根结点。
自定义测试用例:
测试用例的输入由三行组成:
TreeNode root
int fromNode (在 correctBinaryTree 中不可见)
int toNode (在 correctBinaryTree 中不可见)
当以 root 为根的二叉树被解析后,值为 fromNode 的节点 TreeNode 将其右子节点指向值为 toNode 的节点 TreeNode 。然后, root 传入 correctBinaryTree 的参数中。
示例 1:
输入: root = [1,2,3], fromNode = 2, toNode = 3
输出: [1,null,3]
解释: 值为 2 的节点是无效的,所以移除之。
示例 2:
输入: root = [8,3,1,7,null,9,4,2,null,null,null,5,6], fromNode = 7, toNode = 4
输出: [8,3,1,null,null,9,4,null,null,5,6]
解释: 值为 7 的节点是无效的,所以移除这个节点及其子节点 2。
提示:
树中节点个数的范围是 [3, 104] 。
-109 <= Node.val <= 109
所有的 Node.val 都是互不相同的。
fromNode != toNode
fromNode 和 toNode 将出现在树中的同一层。
toNode 在 fromNode 的右侧。
fromNode.right 在测试用例的树中建立后为 null 。
方法1:
主要思路:解题链接汇总
(1)哈希+bfs;
(2)bfs中存储当前结点及其父节点,哈希中存储当前层的各个结点;
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* correctBinaryTree(TreeNode* root) {
unordered_set<TreeNode*> sign;//存储当前层的各个结点
sign.insert(root);
queue<pair<TreeNode*,TreeNode*>>q;//存储当前结点和对应的父节点
q.push({root,nullptr});//初始化
TreeNode*cur=nullptr;
TreeNode*parent=nullptr;
while(!q.empty()){
int cur_size=q.size();//当前层的结点数量
while(cur_size--){
cur=q.front().first;//当前结点
parent=q.front().second;//当前结点的父节点
q.pop();
sign.erase(cur);//更新
if(!q.empty()&&sign.count(cur->right)!=0){//说明当前结点的右子树结点在当前层的结点中,既找到要删除的结点
//删除结点
if(parent->left==cur){
parent->left=nullptr;
}
else{
parent->right=nullptr;
}
return root;
}
//获得下一层的结点
if(cur->left!=nullptr){
q.push({cur->left,cur});
sign.insert(cur->left);
}
if(cur->right!=nullptr){
q.push({cur->right,cur});
sign.insert(cur->right);
}
}
}
return nullptr;
}
};