剑指offer  面试题23:从上往下打印二叉树


参与人数:4853  时间限制:1秒  空间限制:32768K


提交网址: http://www.nowcoder.com/practice/7fe2212963db4790b57431d9ed259701?tpId=13&tqId=11175


题目描述

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


分析:

此题即为二叉树的BFS,使用队列可以解决。


AC代码:

#include<cstdio>
#include<vector>
#include<queue>
using namespace std;

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

};
// 以下为测试 
int main()
{
	Solution sol;
	vector<int> res;
	
	TreeNode *root = new TreeNode(1); 
    root->right = new TreeNode(2); 
    root->right->left = new TreeNode(3); 
	
	res=sol.PrintFromTopToBottom(root);
	
	for(int i:res)
		printf("%d ",i);  // 此处为vector遍历的方法,C++11标准支持 
	return 0;
}