[剑指Offer]从上往下打印二叉树

STL队列的使用

Posted by JinFei on December 17, 2019

题目描述

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

解题思路

  • 考察STL中的队列
  • 写法,queue q,一些基本操作和stack类似,都是push,pop操作,有特殊的front
  • 注意每个入队push的是,当前队列的首元素的左右节点,所以会有一个temp变量来指示。
/*
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;
        if(root == NULL){
            return res;
        }
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty()){
            TreeNode* temp = q.front();
            res.push_back(temp -> val);
            q.pop();
            if(temp -> left){
                q.push(temp -> left);
            }
            if(temp -> right){
                q.push(temp -> right);
            }
        }
        return res;
    }
};