【Leetcode】1660. Correct a Binary Tree

题目地址：

https://leetcode.com/problems/correct-a-binary-tree/

给定一棵二叉树，其中有且只有一个节点的左或者右指针指向了与其同层的节点，要求将该“错误”节点及其子树从原树中删除，返回原树树根。

思路是BFS，同时用哈希表存一下每个节点的父亲。如果遍历某个节点的时候，发现其某个孩子的父亲已经存在了，则说明当前节点指针错误，则找到当前节点的父亲，然后将当前节点删掉。代码如下：

import java.util.HashMap;
import java.util.LinkedList;
import java.util.Map;
import java.util.Queue;

public class Solution {
    public TreeNode correctBinaryTree(TreeNode root) {
        Map<TreeNode, TreeNode> map = new HashMap<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            for (int i = queue.size() - 1; i >= 0; i--) {
                TreeNode cur = queue.poll();
                // 如果其左孩子或者右孩子另有父亲，则说明当前节点错误，将其删去
                if (map.containsKey(cur.left) || map.containsKey(cur.right)) {
                    TreeNode p = map.get(cur);
                    if (cur == p.left) {
                        p.left = null;
                    } else {
                        p.right = null;
                    }
                    
                    return root;
                }
                
                if (cur.left != null) {
                    queue.offer(cur.left);
                    map.put(cur.left, cur);
                }
                if (cur.right != null) {
                    queue.offer(cur.right);
                    map.put(cur.right, cur);
                }
            }
        }
        
        return root;
    }
}

class TreeNode {
    int val;
    TreeNode left, right;
    
    public TreeNode(int val) {
        this.val = val;
    }
}

时空复杂度$O(n)$。