101. 对称二叉树

2022-08-19 14:40:16 浏览数 (1)

101. 对称二叉树

力扣题目链接[1]

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

示例1:

代码语言:javascript复制
输入:root = [1,2,2,3,4,4,3]
输出:true

「提示:」

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

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

思路:

首先使用递归来求解。

递归

首先来看递归终止条件。当前节点的左右子节点都是空,那么当前节点就是对称的,直接返回true。如果当前节点只有左子节点或者右子节点,或者子节点的值不相等,那就是不对称的,直接返回false

然后继续递归处理左左右右和左右右左。直至判断完整个二叉树,亦或者不对称提前返回。

代码语言:javascript复制
/**
 * @param {TreeNode} root
 * @return {boolean}
 */
const recur = (left, right) => {
    if (left === null && right === null) return true;
    if (left === null || right === null || left.val !== right.val) return false;
    return recur(left.left, right.right) && recur(left.right, right.left);
}

var isSymmetric = function(root) {
    return root === null || recur(root.left, root.right);
};

迭代

递归的根本是通过调用栈实现,那么这里可以使用辅助栈来题解。

如果根节点不存在,那么二叉树本身就是对称的。如果根节点不为空,那么将左右子节点同时入栈,方便后续判断。

在循环中,同时弹出左右子节点。

如果左右子节点都存在,则进一步判断值是否相等,如果不相等就意味着不对称,提前返回false。如果相等则继续判断子节点是否对称。做法就是将左左、右右、右左、左右放入栈中,方便下次弹出判断。放入的顺序刚好是左在前,右在后,弹出刚好相反。

如果左节点或者右节点不存在,则意味着不相等,提前返回false

如果遍历完二叉树都没有提前返回,则意味着二叉树对称,返回true

代码语言:javascript复制
/**
 * @param {TreeNode} root
 * @return {boolean}
 */
var isSymmetric = function(root) {
    if (!root) return true;
    let stack = [root.left, root.right];
    while(stack.length) {
        let right = stack.pop();
        let left = stack.pop();
        if (left && right) {
            if (left.val !== right.val) return false;
            stack.push(left.left, right.right, right.left, left.right);
        } else if (left || right) return false;
    }
    return true;
};

总结

本题分别采用递归和迭代来实现对称二叉树的判断。递归需要注意递归的终止条件,迭代需要注意使用栈来存放节点的顺序。

参考资料

[1]

力扣题目链接: https://leetcode-cn.com/problems/symmetric-tree/

0 人点赞