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

1. 题目描述

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

2. 思路解析

  思路就是借助两个辅助单向队列queue来实现，暂且叫他们help1和help2。先把根节点压入help1，然后依次取出help1里面的元素，将其值放入保存结果的vector中，并将其左右节点压入help2，直至help1为空。此时help2中就放满了第二行的所有树节点，然后依次取出help2里面的元素，将其值放入保存结果的vector中，并将其左右节点压入help1，直到help2变为空。
  不停进行以上操作，直至两个辅助单向队列均为空时，就完成了整棵树的遍历。

3. 解答（c++）

/*
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> > treeByLines; // 保存答案的二维vector
            if (pRoot == NULL)  // 判断所给树为空时的特殊情况
                return treeByLines;  // 此时返回的结果也为空

            queue<TreeNode*> help1; // 定义两个辅助单向队列
            queue<TreeNode*> help2;

            help1.push(pRoot); // 向help1中压入根节点

            while((!help1.empty()) || (!help2.empty())) { // 循环直至两个辅助单向队列均变为空
                vector<int> OneLine; // 保存每行的结果

                while(!help1.empty()) {
                    TreeNode* tmp = help1.front();
                    help1.pop();
                    OneLine.push_back(tmp->val);
                    if (tmp->left != NULL)
                        help2.push(tmp->left);
                    if (tmp->right != NULL)
                        help2.push(tmp->right);
                }

                if (!OneLine.empty()) // 如果此行不为空，则压进结果中
                    treeByLines.push_back(OneLine);
                OneLine.clear(); // 清空保存单行结果的vector以进行下一次操作

                while(!help2.empty()) {
                    TreeNode* tmp = help2.front();
                    help2.pop();
                    OneLine.push_back(tmp->val);
                    if(tmp->left != NULL)
                        help1.push(tmp->left);
                    if(tmp->right != NULL)
                        help1.push(tmp->right);
                }
                if (!OneLine.empty())
                    treeByLines.push_back(OneLine);
            }
            return treeByLines;
        }

};