题目描述(简单难度)

反转二叉树,将二叉树所有的节点的左右两个孩子交换。

解法一 递归

对于二叉树的问题,用递归写的话就会异常简单了。交换左右节点,然后左右节点交给递归即可。

public TreeNode invertTree(TreeNode root) {
    if (root == null) {
        return root;
    }

    TreeNode temp = root.left;
    root.left = root.right;
    root.right = temp;

    invertTree(root.left);
    invertTree(root.right);
    return root;
}

解法二 DFS 栈

当然递归都可以用栈模拟,因为解法一的递归比较简单,所以改写也比较容易。

public TreeNode invertTree(TreeNode root) {
    Stack<TreeNode> stack = new Stack<>();
    stack.push(root);
    while (!stack.isEmpty()) {
        TreeNode cur = stack.pop();
        if (cur == null) {
            continue;
        }
        TreeNode temp = cur.left;
        cur.left = cur.right;
        cur.right = temp;

        stack.push(cur.right);
        stack.push(cur.left);
    }
    return root;
}

解法三 BFS 队列

既然可以 DFS,那么也可以 BFS,只需要将解法二的栈改成队列即可。代码不用怎么变,但二叉树的遍历顺序完全改变了。

public TreeNode invertTree(TreeNode root) {
    Queue<TreeNode> queue = new LinkedList<>();
    queue.offer(root);
    while (!queue.isEmpty()) {
        TreeNode cur = queue.poll();
        if (cur == null) {
            continue;
        }
        TreeNode temp = cur.left;
        cur.left = cur.right;
        cur.right = temp;

        queue.offer(cur.left);
        queue.offer(cur.right);
    }
    return root;
}

一道比较简单的题,用递归很快就可以解决。之前一直认为,递归改写成解法二或者解法三的迭代那样会更好一些,因为可以防止递归的堆栈溢出。虽然也有缺点,那就是代码会相对更复杂些,可读性有些降低。

刚才看到 王垠 大神的一个不一样的观点,分享一下。

results matching ""

    No results matching ""