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

力扣热题100之第101题:

方法一、递归:

class Solution {
    public boolean isSymmetric(TreeNode root) {
        //编写函数判断根的左树和右树是否对称
        return func(root.left,root.right);
    }

    public boolean func(TreeNode leftTree,TreeNode rightTree){
        //当左树为空,右树不为空时,不对称!
        if(leftTree == null && rightTree != null){
            return false;
        //当左树不为空,右树为空时,不对称!
        }else if(leftTree != null && rightTree == null){
            return false;
        //当左树右树都为空时,对称!
        }else if(leftTree == null && rightTree == null){
            return true;
        }
        //当左树值不等于右树值时,不对称!
        if(leftTree.val != rightTree.val){
            return false;
        }
        //最后为当左右树都不为空且值相等时,递归判断左树的左和右树的右,
        //左树的右和右树的左对不对称。
        return func(leftTree.left,rightTree.right) && func(leftTree.right,rightTree.left);
    }
}

方法二、迭代:

class Solution {
    public boolean isSymmetric(TreeNode root) {
        Stack<TreeNode> lstack = new Stack<>();
        Stack<TreeNode> rstack = new Stack<>();
        //定义一个左节点,一个右节点
        TreeNode lnode = root.left,rnode = root.right;
        //当左右节点都不为空或者左栈右栈都有元素的情况下,进入循环
        while( lnode != null && rnode != null || !lstack.isEmpty() && !rstack.isEmpty() ){
            int lnum = 0; //左边循环的次数
            while(lnode != null){
                lstack.push(lnode);
                lnode = lnode.left;
                lnum++;
            }
            int rnum = 0; //右边循环的次数
            while(rnode != null){
                rstack.push(rnode);
                rnode = rnode.right;
                rnum++;
            }
            //若左边循环次数不等于右边循环次数,说明左边或右边少了结点,即为不对称!
            if(lnum != rnum){
                return false;
            }
            TreeNode lret = lstack.pop();
            TreeNode rret = rstack.pop();
            //若左栈右栈弹出的节点值不相等,则不对称!
            if(lret.val != rret.val){
                return false;
            }
            lnode = lret.right;
            rnode = rret.left;
        }
        //若循环退出,左节点或右节点有为空的,说明不对称!
        if(lnode == null && rnode != null){
            return false;
        }else if(lnode != null && rnode == null){
            return false;
        }
        return true;
    }
}

使用迭代方法时,需要用到栈来辅助解答,新建一个左栈用来存放左子树的节点、一个右栈用来存放右子树的节点。

猜你喜欢

转载自blog.csdn.net/weixin_56960711/article/details/123338018