代码随想录算法训练营第十五天|102. 二叉树的层序遍历、226. 翻转二叉树、101. 对称二叉树

Leetcode 102. 二叉树的层序遍历
思路：逐层遍历树，用一个队列来保存遍历到的元素，遍历顺序是前序遍历，每遍历一层都会根据队列的长度更新遍历的次数，用一个数组存储每一层遍历到的元素

/**
 * 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:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> que; //定义一个队列来存放遍历到的元素
        if(root != NULL)que.push(root);
        vector<vector<int>> res;
        while(!que.empty()){ 
            int size = que.size(); //单独记录队列长度 因为弹出元素后数列长度会变
            vector<int> vec; //存放每一层遍历到的元素
            while(size--){
                TreeNode* node = que.front(); //逐层遍历树
                que.pop();
                vec.push_back(node->val);
                if(node->left)que.push(node->left); //左节点不为空遍历左节点
                if(node->right)que.push(node->right); //右节点不为空遍历右节点
            }
            res.push_back(vec);
        }
        return res;
    }
};

Leetcode 226. 翻转二叉树
思路：核心是实现左右节点的交换，不能用中序遍历的顺序遍历树，可以用前序遍历和后序遍历。因为中序节点某些节点的左右子节点会被交换两次

/**
 * 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:
    TreeNode* invertTree(TreeNode* root) {
        if(root==NULL)return root; //递归出口
        swap(root->left,root->right); //中
        invertTree(root->left); //左
        invertTree(root->right); //右
        return root;
    }
};

Leetcode 101. 对称二叉树
思路：这一道题不是比较左右节点，而是比较左右子树是否可以翻转，此时只能使用后序遍历，因为我们要保存左右节点的信息，判断左右子树是否可以翻转。判断整个树的外侧节点和内侧节点是否相等，也就是判断左子树的左节点和右子树的右节点，和左子树的右节点以及右子树的左节点是否相等

/**
 * 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 compare(TreeNode* left,TreeNode* right){ //比较根节点的左右子树是否可以相互翻转
        if(left==NULL && right==NULL)return true; //都为空指针
        else if(left == NULL && right != NULL || left != NULL && right == NULL)return false; //有一个为空指针
        else if(left->val != right->val)return false; //两个节点的值不相等
        bool outside = compare(left->left, right->right); //左节点的左侧和右节点的右侧
        bool inside = compare(left->right,right->left); //左节点的右侧和右节点的左侧
        return (outside && inside); //都为true结果才为true
    }
    bool isSymmetric(TreeNode* root) {
        if(root==NULL)return true;
        return compare(root->left,root->right);
    }
};