剑指offer——JZ32 从上往下打印二叉树 解题思路与具体代码【C++】

一、题目描述与要求

从上往下打印二叉树_牛客题霸_牛客网 (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;
    }
};