题目描述
描述:不分行从上往下打印出二叉树的每个节点,同层节点从左至右打印。例如输入{8,6,10,#,#,2,1},如以下图中的示例二叉树,则依次打印8,6,10,2,1(空节点不打印,跳过),请你将打印的结果存放到一个数组里面,返回。
数据范围:
0<=节点总数<=1000;
-1000<=节点值<=1000;
输入:{8,6,10,#,#,2,1}
返回值:[8,6,10,2,1]
输入:{5,4,#,3,#,2,#,1}
返回值:[5,4,3,2,1]
解题思路
从上往下打印二叉树:最直观的想法是,层次遍历,使用辅助队列进行层次遍历。使用res存储遍历结果,使用que辅助存储二叉树,使用cur表示当前队列的头节点。首先判断根结点root是否为空,如果是则直接返回空的res,反之将头节点加入到队列中,当队列不为空,就使用cur接收队列的头节点,然后将头节点弹出来,并且当头结点的左孩子不为空则将左孩子加入队列,头孩子的右节点不为空则将右孩子加入队列,接着将头节点的值加入结果数组中,最后返回res即可。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode* root) {
vector<int> res;
queue<TreeNode *> que;
TreeNode *cur;
if(!root) return res;
que.push(root);
while(!que.empty())
{
cur=que.front();
que.pop();
if(cur->left)que.push(cur->left);
if(cur->right)que.push(cur->right);
res.push_back(cur->val);
}
return res;
}
};
前中后序即深度优先搜索我喜欢写递归,层序遍历即广度优先搜索我喜欢写队列。