LeetCode 101 对称二叉树

题目描述

对称二叉树

给你一个二叉树的根节点 root ， 检查它是否轴对称。

示例 1：

输入：root = [1,2,2,3,4,4,3]
输出：true

示例 2：

输入：root = [1,2,2,null,3,null,3]
输出：false

提示：

• 树中节点数目在范围 [1, 1000] 内
• -100 <= Node.val <= 100

**进阶：**你可以运用递归和迭代两种方法解决这个问题吗？

解法

解法1：递归

从根节点出发，分别比较左子树和右子树

java代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public boolean isSymmetric(TreeNode root) {
        if (root == null) {
            return false;
        }

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

        return check(left, right);

    }

    private boolean check(TreeNode left, TreeNode right) {
        if (left == null && right == null) {
            return true;
        }

        if (left != null && right != null) {
            if (left.val != right.val) {
                return false;
            }
            return check(left.right, right.left) & check(left.left, right.right);
        }
        return false;
    }
}

复杂度

• 时间复杂度：O(n) ，其中 n 为二叉树的节点个数。在递归调用的时候二叉树的每个节点最多被访问一次。
• 空间复杂度：O(n)，递归函数在递归过程中使用栈空间有关，这里递归层数不超过n

解法2：迭代使用栈模拟中序遍历。

java代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public boolean isSymmetric(TreeNode root) {
        return check(root, root);
    }


    private boolean check(TreeNode left, TreeNode right) {
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        q.offer(left);
        q.offer(right);

        while (!q.isEmpty()) {
            right = q.poll();
            left = q.poll();

            // 如果都为空，则继续遍历
            if (right == null && left == null) {
                continue;
            }

            // 有一个为null。或值不相等
            if ((left == null || right == null) || (left.val != right.val)) {
                return false;
            }

            // 把左节点的左节点和右节点的右节点入栈
            q.offer(left.left);
            q.offer(right.right);

            // 把左节点的右节点和右节点的左节点入栈
            q.offer(left.right);
            q.offer(right.left);
        }
        return true;
    }
}

复杂度

• 时间复杂度：O(n)
• 空间复杂度：O(n)，其中 n 为二叉树的节点个数。栈最多存储 n 个节点，因此需要额外的 O(n) 的空间。