面试题32 - Ⅰ 从上到下打印二叉树

题目地址

解题思路

这道题要实现二叉树的层序打印，所以想到利用队列来保存节点。因为队列具有先进先出的特点，当我们对一个树节点进行操作时，可以判断其左右子节点是否存在。

这样就完美形成了一个迭代的闭环。

感觉代码写着写着，变成了代码随想录卡哥的形状，可能也是模仿多了吧。

代码实现（C++）

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> levelOrder(TreeNode* root) 
    {
        vector<int> res;
        queue<TreeNode*> que;
        if(root==nullptr)
        {
            return res;
        }
        que.push(root);
        while(!que.empty())
        {
            TreeNode* node=que.front();
            que.pop();
            res.push_back(node->val);
            
            if(node->left)
            {
                que.push(node->left);
            }
            if(node->right)
            {
                que.push(node->right);
            }
        }
        return res;
    }
};