【二叉树算法题记录】101. 对称二叉树

题目描述

题目链接
给你一个二叉树的根节点 root， 检查它是否轴对称。

题目分析

【迭代】层序遍历

轴对称可以看作是每一层结点头尾对称（相等）。那么层序遍历就直接上了：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool isSymmetric(TreeNode* root) {
        // 对称实际上可以演变成每一层头尾对应的结点value是一样的
        // 可以用层序遍历
        queue<TreeNode*> q;
        //vector<vecotr<int>> all;
        if(root!=NULL) q.push(root);
        while(!q.empty()) {
            int size = q.size();
            vector<int> vec;
            while(size--) { // 遍历每层，然后记录他们的value到vec数组
                TreeNode* node = q.front(); q.pop();
                if(node) vec.push_back(node->val);
                else vec.push_back(404); // 如果结点不存在，就记录他的值为404

                if(node) {
                    q.push(node->left);
                    q.push(node->right);
                }
            }

            // 对vec数组进行判断
            for(int i = 0, j = vec.size()-1; i < j; i++, j--) {
                cout<< vec[i] << " " << vec[j] << endl;
                if(vec[i]!=vec[j]) return false;
            }
        }
        return true;
    }
};

注意这里每层遍历时还要记录不存在的结点，不然会出错。

【递归】

递归通过比较左右子树翻转是否相同来判断，那么就得采取后序遍历的方式，因为我们首先要收集左右孩子的信息传给上层结点，所以要先访问左右孩子节点，也就是后序遍历。

构造递归的三个步骤：

• 确定递归函数参数：因为我们比较的左右子树，所以应该同时传入根节点的左右孩子，即bool compare(TreeNode* left, TreeNode* right)
• 确定递归终止条件：这里我们要考虑到多种情况，包括
  （1）左孩子为空，右孩子也为空（left == NULL && right == NULL），对称（true）；
  （2）左孩子为空，右孩子不为空（left == NULL && right != NULL），不对称（false）；
  （3）左孩子不为空，右孩子为空（left != NULL && right == NULL），不对称（false）；
  （4）左孩子右孩子都不为空，但是值不相等（left->val != right->val），不对称（false）。
• 确定递归逻辑：我们比较左右子树有两条路径，即left->left与right->right，left->right与right->left，所以我们的递归有两条路线：compare(left->left, right->right)与compare(left->right, right->left)。他们比较出来都是true才是对称的子树。

递归法的整体代码如下：

class Solution {
public:
    // 确定递归参数
    bool compare(TreeNode* left, TreeNode* right) {
        // 确定递归终止条件
        // 首先排除空节点的情况
        if (left == NULL && right != NULL) return false;
        else if (left != NULL && right == NULL) return false;
        else if (left == NULL && right == NULL) return true;
        // 排除了空节点，再排除数值不相同的情况
        else if (left->val != right->val) return false;


        // 确定递归逻辑
        bool comp_1 = compare(left->left, right->right);
        bool comp_2 = compare(left->right, right->left);
        bool comp = comp_1 && comp_2;
        return comp;
    }

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