Leetcode Symmetric Tree

weixin_33774883

于 2017-05-21 12:15:00 发布

阅读量47

点赞数

本文介绍了一种判断二叉树是否对称的方法，包括非递归和递归两种解法。通过栈来实现非递归解法，递归解法则直接比较左右子树是否镜像对称。

非递归解法

/**
 * 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) {
       stack <TreeNode*> stackLeft;
       stack <TreeNode*> stackRight;
       
       if(root == NULL ||(root && (!root->left) && (!root->right))){
           return true;
       }
       
       if((root->left && (!root->right))||(!root->left && root->right)){
           return false;
       }
       TreeNode *pLeft = root->left;
       TreeNode *pRight = root->right;
       while((pLeft||(!stackLeft.empty())) && (pRight ||(!stackRight.empty()))){
            while(pLeft && pRight){
               
                if(pLeft->val != pRight->val){
                    return false;
                }
                stackLeft.push(pLeft); 
                pLeft = pLeft->left;
                stackRight.push(pRight);
                pRight = pRight->right;
            }
            
            if(pLeft || pRight){
                return false;
            }
            
            pLeft = stackLeft.top();
            stackLeft.pop();
            pLeft = pLeft->right;
            
            pRight = stackRight.top();
            stackRight.pop();
            pRight = pRight->left;
       }
       
       if( (pLeft||(!stackLeft.empty())) || (pRight ||(!stackRight.empty()))){
           return false;
       }
       
       return true;
    }
};<pre name="code" class="cpp">

递归解法

/**
 * 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 == NULL || (root && !root->left && !root->right)){
           return true;
       }
       
       if( (root->left && !root->right) || (!root->left && root->right) ) {
           return false;
       }
       
       return checkLeftAndRight(root->left, root->right);
    }
    
    bool checkLeftAndRight(TreeNode *pLeft, TreeNode *pRight){
        
        if(!pLeft && !pRight){
            return true;
        }
        
        if( (pLeft && !pRight) || (!pLeft && pRight)){
            return false;
        }
        
        if(pLeft->val == pRight->val){
            return checkLeftAndRight(pLeft->left, pRight->right) && checkLeftAndRight(pLeft->right, pRight->left);
        } else {
            return false;
        }
    }
};