[LeetCode]Symmetric Tree

最新推荐文章于 2019-05-29 13:59:21 发布

原创最新推荐文章于 2019-05-29 13:59:21 发布 · 193 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树

LeetCode 专栏收录该内容

65 篇文章

订阅专栏

本文介绍了一种检查二叉树是否为中心对称的方法，提供了递归与非递归两种实现方式。通过实例说明了如何判断一棵二叉树是否与其镜像相同。

Given a binary tree, check whether it is a mirror of itself (ie, symmetric around its center).

For example, this binary tree is symmetric:

But the following is not:

Note:
Bonus points if you could solve it both recursively and iteratively.

解法是：画一个更大的图来找规律，然后就知道如何递归的比较了。

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isSymmetric(TreeNode *a, TreeNode *b)
    {
        if (!a && !b) return true;
        else if (!a || !b) return false;
        
        if (a->val != b->val) return false;
        return isSymmetric(a->left, b->right) && isSymmetric(a->right, b->left);
    }

    bool isSymmetric(TreeNode *root) {
        if (!root) return true;
        return isSymmetric(root->left, root->right);
    }
};

非递归版就是模仿一下递归版，在转化为非递归的时候，我们发现每次需要保存的“状态”就是两个待比较的树的根，因此我们的栈操作每次都压入两个、弹出两个，成双成对就可以了。

/**
 * Definition for binary tree
 * 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) return true;
        stack<TreeNode*> s;
        s.push(root->right); // right tree root
        s.push(root->left);  // left tree root
        while (!s.empty())
        {
            TreeNode *left = s.top(); s.pop();
            TreeNode *right = s.top(); s.pop();
            if (!left && !right) continue;
            else if (!left || !right) return false;
            if (left->val != right->val) return false;
            s.push(right->right);
            s.push(left->left);
            s.push(right->left);
            s.push(left->right);
        }
        return true;
    }
};