欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【每日刷题】对称二叉树

程序员文章站 2022-05-16 21:29:40
...

题目地址

https://leetcode-cn.com/problems/symmetric-tree/

题目描述:对称二叉树

给定一个二叉树,检查它是否是镜像对称的。

例如,二叉树 [1,2,2,3,4,4,3] 是对称的。

    1
   / \
  2   2
 / \ / \
3  4 4  3

但是下面这个 [1,2,2,null,3,null,3] 则不是镜像对称的:

    1
   / \
  2   2
   \   \
   3    3

解答

阅读理解题。

我们用层次遍历的方法来检测。

利用队列进行迭代。队列中每两个连续的结点应该是相等的,而且它们的子树互为镜像。最初,队列中包含的是 root->left 以及 root->right。该算法的工作原理类似于 BFS,但存在一些关键差异。每次提取两个结点并比较它们的值。然后,将两个结点的左右子结点按相反的顺序插入队列中。当队列为空时,或者我们检测到树不对称(即从队列中取出两个不相等的连续结点)时,该算法结束。

代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isSymmetric(TreeNode* root) {
        if( !root || (!root->left && !root->right)) //当不存在结点或只有一个结点时候为真
            return true;
        
        queue<TreeNode*> left, right;
        
        if( root->left && root->right)
            left.push( root->left), right.push( root->right);
        else
            return false;
        
        while( !left.empty() || !right.empty()){
            if( left.empty() || right.empty())
                return false;

            TreeNode* leftTemp = left.front(), * rightTemp = right.front();
            left.pop(), right.pop();

            if( leftTemp->val != rightTemp->val)
                return false;

            if( leftTemp->left && rightTemp->right)
                left.push( leftTemp->left), right.push( rightTemp->right);
            else if( (leftTemp->left && !rightTemp->right) || (( !leftTemp->left && rightTemp->right)))
                return false;


            if( leftTemp->right && rightTemp->left)
                left.push( leftTemp->right), right.push( rightTemp->left);
            else if( (leftTemp->right && !rightTemp->left) || ( !leftTemp->right && rightTemp->left))
                return false;
        }
        return true;
    }
};