leetcode——102.二叉树的层序遍历

思路

1. 利用广度优先搜索，维护一个队列
2. 将树根放入队列，取出，访问其左右孩子，使其入队
3. 按照层次遍历——按照当前结点数量进行循环压入下一层结点

代码

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector <vector <int>> ret;
        if (!root) 
            return ret;

        queue <TreeNode*> q;
        q.push(root);
        while (!q.empty()) {
            int currentLevelSize = q.size();//获取每层结点个数 1 2 4 8...
            ret.push_back(vector <int> ());//每层申请一个向量存储
            for (int i = 1; i <= currentLevelSize; ++i) {
                auto node = q.front(); q.pop();//获取队列头
                ret.back().push_back(node->val);//访问并记录——插入当前二维向量最后一个
                if (node->left) q.push(node->left);//压入左孩子
                if (node->right) q.push(node->right);//压入右孩子
            }
        }
        return ret;
    }
};