直通BAT算法精讲课：1.2 二叉树打印练习题

有一棵二叉树，请设计一个算法，按照层次打印这棵二叉树。

给定二叉树的根结点root，请返回打印结果，结果按照每一层一个数组进行储存，所有数组的顺序按照层数从上往下，且每一层的数组内元素按照从左往右排列。保证结点数小于等于500。

#include<iostream>
#include<vector>
#include<queue>
using namespace std;
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};

class TreePrinter {
public:
    vector<vector<int> > printTree(TreeNode* root) {
      	vector<vector<int> >result;
		
        queue<TreeNode *>q;
        if(root==NULL) return result;
        q.push(root);
        while(!q.empty())
        {

            int size =q.size();
            vector<int> row;
            for(int i=0;i<size;i++)  //size 每次保存的是 层次遍历中每一行的节点个数  
            {
            	 TreeNode *save=q.front();//务必将队列的头结点在此保存下来 
                 row.push_back(save->val);
                 q.pop();//因为在这里需要出队列  如果还是用q.front()  那么将死循环 因为之前的根节点已经出队列了 
                 if(save->left!=NULL)
                {
                    q.push(save->left);
                }
                if(save->right!=NULL)
                {
                    q.push(save->right);
                }
            }
            result.push_back(row);
        }
        
        return result;
    }
};

int main()
{
	TreePrinter tre;
	TreeNode *root=new TreeNode(1);
	root->left=new TreeNode(2);
	root->right=new TreeNode(3);
	root->left->left=new TreeNode(4);
	
	vector<vector<int> > res;
	res=tre.printTree(root);
	for(int i=0;i<res.size();i++)
	{
		
		for(int j=0;j<res[i].size();j++)
		{
			cout<<res[i][j]<<" ";
		}
		cout<<endl;
	}
	/*
		打印出
		1
		2 3
		4
		测试成功 
	*/ 
	
}