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

一、题目描述与要求

从上往下打印二叉树_牛客题霸_牛客网 (nowcoder.com)

题目描述

不分行从上往下打印出二叉树的每个节点,同层节点从左至右打印。例如输入{8,6,10,#,#,2,1},如以下图中的示例二叉树,则依次打印8,6,10,2,1(空节点不打印,跳过),请你将打印的结果存放到一个数组里面,返回。

剑指offer——JZ32 从上往下打印二叉树 解题思路与具体代码【C++】_第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即可。

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


三、具体代码

class Solution {
public:
    vector PrintFromTopToBottom(TreeNode* root) {
		vector v;
		if(root==nullptr) return v;//如果是空树则返回空vector
		queue 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;
    }
};

你可能感兴趣的:(剑指offer,c++,算法,数据结构)