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

层序遍历 10

文章
视频

迭代法：队列存每一层的节点，后续加入的下一层孩子节点下一次循环处理，
因此关键是要记录每一层的数量

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<vector<int>> result;
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            // 这里一定要使用固定大小size，不要使用que.size()，因为que.size是不断变化的
            for (int i = 0; i < size; i++) {
                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);
            }
            result.push_back(vec);
        }
        return result;
    }
};

递归法：深度信息

class Solution {
public:
    void order(TreeNode* cur, vector<vector<int>>& result, int depth)
    {
        if (cur == nullptr) return;
        if (result.size() == depth) result.push_back(vector<int>());
        result[depth].push_back(cur->val);
        order(cur->left, result, depth + 1);
        order(cur->right, result, depth + 1);
    }
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> result;
        int depth = 0;
        order(root, result, depth);
        return result;
    }
};

226.翻转二叉树

题目

文章
视频

只要把每一个节点的左右孩子翻转一下，就可以达到整体翻转的效果

递归法

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;
    }
};

迭代法
迭代法统一写法
层序遍历

101.对称二叉树 2

题目
文章讲解
视频
外侧和内侧分别相等，确认遍历顺序（前中后序）

class Solution {
public:
    bool Compare(TreeNode* left,TreeNode* right)
    {
        if(left==right&&left==NULL)
            return true;
        else if(left&&right&&left->val==right->val)
        {
            return(Compare(left->left,right->right)&&Compare(left->right,right->left));
        }
        else
            return false;
    }
    bool isSymmetric(TreeNode* root) {
        return Compare(root->left,root->right);
    }
};

比较简洁，原来这就是后序。

还可以用迭代法，就是左右子树的内外侧放入堆栈或队列，两两比较并继续放入