剑指Offer 层序遍历二叉树

原地址

流程：
特例处理： 当根节点为空，则返回空列表 [] ；
初始化： 打印结果列表 res = [] ，包含根节点的队列 queue = [root] ；
BFS 循环： 当队列 queue 为空时跳出；
新建一个临时列表 tmp ，用于存储当前层打印结果；
当前层打印循环： 循环次数为当前层节点数（即队列 queue 长度）；
出队： 队首元素出队，记为 node；
打印： 将 node.val 添加至 tmp 尾部；
添加子节点： 若 node 的左（右）子节点不为空，则将左（右）子节点加入队列 queue ；
将当前层结果 tmp 添加入 res 。

 

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {

if(root == nullptr) return  vector<vector<int>>();
        vector<vector<int>> vecRet;

        queue<TreeNode *>que;
        que.push(root);

        while(!que.empty()) {
            int size = que.size();
            vector<int>vecTemp;

            for(int i = 0; i < size; i++) {
                //依次弹出并存值在temp数组中
                 auto node = que.front();
                 que.pop();
                 vecTemp.push_back(node->val);

                 
                if(node->left){
                    que.push(node->left);
                }
                if(node->right){
                    que.push(node->right);
                }
            }
            vecRet.push_back(vecTemp);
        }
        return vecRet;

    }
};