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

剑指offer题型分类及各题的代码及解题思路

1、题目描述

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

2、初始结构定义如下：

/*
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) {}
};

3、思路

本题考查树的层次遍历，我们可以借助一个queue和vector来实现：

① 先判断根节点是否为空，若为空，则返回，否则将根节点入队；

② 队列不为空时，取队列的首元素，将其放入vector；

③ 再遍历其子节点，将子节点左右指针放入队列中；

④ 重复第②、③步，直至队列为空；

⑤ 返回vector。

4、C++实现代码如下

 

class Solution 
{
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) 
    {
        vector<int> res;
        
        if (!root)
            return res;
        
        queue<TreeNode*> q;
        q.push(root);
        while (!q.empty())
        {
            TreeNode* tmp = q.front();
            q.pop();
            res.push_back(tmp->val);
            
            if (tmp->left)
                q.push(tmp->left);
            
            if (tmp->right)
                q.push(tmp->right);
        }
        return res;
    }
};