剑指Offer面试题32 - I. 从上到下打印二叉树

Posted 2021-03-07 galaxy-hao

题目

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

例如:
给定二叉树:?[3,9,20,null,null,15,7],

    3
   /   9  20
    /     15   7

返回：
[3,9,20,15,7]

提示：
节点总数 <= 1000

思路

借助队列简单层次遍历。

代码

时间复杂度：O(n)
空间复杂度：O(n)

class Solution {
public:
    vector<int> levelOrder(TreeNode* root) {
        vector<int> res;
        if (root) {
            queue<TreeNode*> que;
            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;
    }
};