纠正二叉树

/*
 题目思路: 
 由于Java Stream API不适合直接操作和修改数据结构，因此我们在这里仍然使用经典的遍历方法。
 使用Stream API的部分是为了展示我们如何使用它来进行一些集合操作，例如过滤和查找。
 我们通过遍历树，记录看到的节点，并处理发现的误入节点。
*/
public class Solution {
    public TreeNode correctBinaryTree(TreeNode root) {
        Set<TreeNode> seen = new HashSet<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            Stream.of(node.right, node.left).filter(Objects::nonNull).forEach(child -> {
                if (seen.contains(child)) {
                    return null;
                }
                queue.add(child);
            });
            seen.add(node);
        }
        return root;
    }
}

/* 
假设树节点类如下：
public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}
*/