剑指offer22_从上往下打印二叉树

从上往下打印二叉树

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

示例1

输入

{5,4,#,3,#,2,#,1}

返回值

[5,4,3,2,1]

分析

方案一：宽度优先搜索

/**
时间复杂度 O(N) ： N 为二叉树的节点数量，即 BFS 需循环 N 次。
空间复杂度 O(N) ： 最差情况下，即当树为平衡二叉树时，最多有 N/2 个树节点同时在 queue 中，使用 O(N) 大小的额外空间。
**/
class Solution
{
public:
    vector<int> PrintFromTopToBottom(TreeNode *root)
    {
        vector<int> ans;
        // 当树的根节点为空，则直接返回空列表
        if(!root) return ans;
        queue<TreeNode *> q;
        q.push(root);
        // BFS循环
        while (!q.empty())
        {
            TreeNode *top = q.front();
            q.pop();//队首元素出队
            ans.push_back(top->val);
            //若top的左右子节点不为空，则将左右子节点加入队列
            if (top->left) q.push(top->left);
            if (top->right) q.push(top->right);
        }
        return ans;
    }
};