102.二叉树层序遍历（队列）107.二叉树的层序遍历II（102的反转）

102.二叉树层序遍历（队列）

解题思路

使用队列实现，队列变化的size为返回的二维数组每行的大小

代码

/**
 * 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>> result;
        //队列不为空时
        while(!que.empty()){
            //由于队列的长度在不断变化，获取队列size作为返回数组的行大小
            int size = que.size();
            vector<int> line;   //存放每层的结果
            
            for(int i=0;i<size;i++){
                TreeNode* node = que.front();   //取到队列第一个node
                que.pop();  //出队
                line.push_back(node->val);  
                // 若存在左右子树， 子树根节点入栈
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
            //for 循环结束，即处理完一行，装入结果中
            result.push_back(line);
        }
        return result;
    }
};

107.二叉树的层序遍历II（102的反转）

解题思路

对102题的结果进行翻转，reverse()

代码

/**
 * 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>> levelOrderBottom(TreeNode* root) {
queue<TreeNode*> que;
        if(root != NULL) que.push(root);    //根节点入队
        vector<vector<int>> result;
        //队列不为空时
        while(!que.empty()){
            //由于队列的长度在不断变化，获取队列size作为返回数组的行大小
            int size = que.size();
            vector<int> line;   //存放每层的结果
            
            for(int i=0;i<size;i++){
                TreeNode* node = que.front();   //取到队列第一个node
                que.pop();  //出队
                line.push_back(node->val);  
                // 若存在左右子树， 子树根节点入栈
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
            //for 循环结束，即处理完一行，装入结果中
            result.push_back(line);
        }
        reverse(result.begin(),result.end());//翻转数组
        return result;
    }
};