leetcode-二叉树的层序遍历-77

Posted 2021-10-28 天津 唐秙

题目要求
  给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。
解析
  广度优先搜索，入队出队
代码实现

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* t) {
        vector<vector<int>> ret;
        if(t == nullptr)
            return ret;

        queue <TreeNode*> q;
        q.push(t);
        while(!q.empty())
        {
            int size = q.size();
            ret.push_back(vector<int>());
            for(int i = 1; i <= size; 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;
    }
};