JZTXT

分行从上往下打印二叉树

发布时间 2023-03-31 09:20:40作者: 穿过雾的阴霾
class Solution {
public:
    vector<vector<int>> res;
    void bfs(TreeNode* root)
    {
        queue<TreeNode*> q;
        q.push(root);
        while(q.size ())
        {
            int size=q.size();
            vector<int> tmp;
            while(size--)
            {
                auto p=q.front();
                q.pop();
                tmp.push_back(p->val);
                if(p->left) q.push(p->left);
                if(p->right) q.push(p->right);
            }
            res.push_back(tmp);
        }
    }
    vector<vector<int>> printFromTopToBottom(TreeNode* root) {
        if(!root)   return res;
        bfs(root);
        return res;
    }
};