剑指offer——把二叉树打印成多行

题目描述

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

 

不要求每行反序，比上一题还要简单。

class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int> > result;
            if(pRoot == NULL)
                return result;
            list<TreeNode*> pNode1, pNode2;
            vector<int> temp;
            pNode1.push_back(pRoot);
            while(!pNode1.empty() || !pNode2.empty())
            {
                while(!pNode1.empty())
                {
                    TreeNode* node = pNode1.front();
                    pNode1.pop_front();
                    temp.push_back(node->val);
                    if(node->left != NULL)
                        pNode2.push_back(node->left);
                    if(node->right != NULL)
                        pNode2.push_back(node->right);
                }
                if(!temp.empty())
                {
                    result.push_back(temp);
                    temp.clear();
                }
                
                while(!pNode2.empty())
                {
                    TreeNode* node = pNode2.front();
                    pNode2.pop_front();
                    temp.push_back(node->val);
                    if(node->left != NULL)
                        pNode1.push_back(node->left);
                    if(node->right != NULL)
                        pNode1.push_back(node->right);
                }
                if(!temp.empty())
                {
                    result.push_back(temp);
                    temp.clear();
                }
            }
            return result;
        }
    
};

如果不要求按层打印，只要求按顺序打印的话，那么就可以只用一个 vector<TreeNode*> ，前面出，后面放。就更简洁了。