226.翻转二叉树

最新推荐文章于 2024-03-08 13:33:16 发布

wangchengxi03

最新推荐文章于 2024-03-08 13:33:16 发布

阅读量109

点赞数

分类专栏：刷题题解文章标签： leetcode 算法

本文链接：https://blog.csdn.net/m0_51321956/article/details/126896824

版权

刷题题解专栏收录该内容

8 篇文章 0 订阅

订阅专栏

226. 翻转二叉树

递归法

class Solution {
    public TreeNode invertTree(TreeNode root) {
        // 递归终止条件
        if (root == null) {
            return root;
        }
        // 交换当前节点的左右子节点，这一步可以放在递归调用root.left和root.right之后
        // 不能放在中间，因为你已经将左右子节点交换，再递归调用的话，相当于只是把根节点的左右子节点交换而已
        TreeNode temp = root.left;
        root.left = root.right;
        root.right = temp;
        // 下面这两次递归调用顺序可以不一样
        invertTree(root.left);
        invertTree(root.right);
        return root;
    }
}

队列实现

class Solution {
    public TreeNode invertTree(TreeNode root) {
        if (root == null) {
            return root;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            // 交换左右子节点的顺序可在前可在后
            TreeNode temp = node.left;
            node.left = node.right;
            node.right = temp;
            // 下面node.left和node.right谁先进队无所谓
            if (node.left != null) {
                queue.add(node.left);
            }
            if (node.right != null) {
                queue.add(node.right);
            }
        }
        return root;
    }
}

栈实现

class Solution {
    public TreeNode invertTree(TreeNode root) {
        if (root == null) {
            return root;
        }
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();
            // 交换左右子节点的顺序可在前可在后
            TreeNode temp = node.left;
            node.left = node.right;
            node.right = temp;
            // 下面node.left和node.right谁先进栈无所谓
            if (node.left != null) {
                stack.add(node.left);
            }
            if (node.right != null) {
                stack.add(node.right);
            }
        }
        return root;
    }
}