利用队列进行二叉树的层序遍历

leetcode 102. 二叉树的层序遍历

题目链接

给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。
示例：
二叉树：[3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层序遍历结果：
[
[3],
[9,20],
[15,7]
]

思路

利用一个队列每次保存一层的节点，然后再遍历该层节点的时候将其非空的左右孩子加入队列，最后队列为空则层序遍历完成

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

    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> queNode;
        vector<vector<int>> result;
        if(root == nullptr)
            return result;
        queNode.push(root);
        while(!queNode.empty()){
            int size = queNode.size();		//计算当前层的节点数
            vector<int> res(size);
            for(int i = 0; i < size; ++i){
                TreeNode* tmpNode = queNode.front();
                res[i] = tmpNode->val;
                if(tmpNode->left != nullptr)
                    queNode.push(tmpNode->left);
                if(tmpNode->right != nullptr)
                    queNode.push(tmpNode->right);
                queNode.pop();
            }
            result.push_back(res);
        }
        return result;
    }