把二叉树打印成多行

题目描述

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

方法一

中序遍历，使用队列，将每一行的元素依次压入队列，每将队列中元素弹出一个就将其左右子节点压入队列。在每一行的开始节点时，队列中元素的个数就是下一行节点的个数，通过计数取出该行所有元素。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int> >res;
            if(pRoot==NULL)
                return res;
            queue<TreeNode*>q;
            TreeNode*point;
            q.push(pRoot);
            vector<int>tmp;
            while(!q.empty())
            {
                int l=0;
                int len=q.size();
                while(l++<len)
                {
                    point=q.front();
                    q.pop();
                    tmp.push_back(point->val);
                    if(point->left!=NULL)
                        q.push(point->left);
                    if(point->right!=NULL)
                        q.push(point->right);
                }
                res.push_back(tmp);
                tmp.clear();
            }
            return res;
        }
};

方法二

使用两个队列，第一行元素压入队列1，打印第一行元素时，将第二行元素压入队列2，打印第二行元素时，将第三行元素压入队列1…，直到两个队列都为空。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int> >res;
            if(pRoot==NULL)
                return res;
            queue<TreeNode*>q1;
            queue<TreeNode*>q2;
            TreeNode*point;
            q1.push(pRoot);
            vector<int>tmp;
            while(!q1.empty()||!q2.empty())
            {
                while(!q1.empty())
                {
                    point=q1.front();
                    tmp.push_back(point->val);
                    q1.pop();
                    if(point->left!=NULL)
                        q2.push(point->left);
                    if(point->right!=NULL)
                        q2.push(point->right);
                }
                if(!tmp.empty())
                {
                    res.push_back(tmp);
                    tmp.clear();
                }
                while(!q2.empty())
                {
                    point=q2.front();
                    tmp.push_back(point->val);
                    q2.pop();
                    if(point->left!=NULL)
                        q1.push(point->left);
                    if(point->right!=NULL)
                        q1.push(point->right);
                }
                if(!tmp.empty())
                {
                    res.push_back(tmp);
                    tmp.clear();
                }
            }
            return res;
        }
};