一、题目描述与要求
从上往下打印二叉树_牛客题霸_牛客网 (nowcoder.com)
题目描述
不分行从上往下打印出二叉树的每个节点,同层节点从左至右打印。例如输入{8,6,10,#,#,2,1},如以下图中的示例二叉树,则依次打印8,6,10,2,1(空节点不打印,跳过),请你将打印的结果存放到一个数组里面,返回。
数据范围:
0<=节点总数<=1000
-1000<=节点值<=1000
示例
示例1:
输入:{8,6,10,#,#,2,1}
返回值:[8,6,10,2,1]
示例2:
输入:{5,4,#,3,#,2,#,1}
返回值:[5,4,3,2,1]
二、解题思路
根据题目描述从上到下打印二叉树其实就是层次遍历二叉树,利用队列实现即可。
首先判断是否为空树,是的话直接返回空vecot;
先让根节点入队,然后当队列不为空的时候,出队列,然后把出队列的结点的值存入vector,然后判断当前结点的左子树与右子树是否为空,不为空则压入队列。最后返回v即可。
三、具体代码
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode* root) {
vector<int> v;
if(root==nullptr) return v;//如果是空树则返回空vector
queue<TreeNode*> q;//利用队列进行层次遍历
TreeNode* p;
q.push(root);//先让根节点入队列
while(!q.empty()){ //当队列不为空
p=q.front();
q.pop();//出队列
v.push_back(p->val);
if(p->left) q.push(p->left);
if(p->right) q.push(p->right);
}
return v;
}
};