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

题目描述

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

 

解题思路：

二叉树层次遍历, 利用一个队列，每层之间用一个NULL分割开，当遇到NULL则代表当前层已遍历完

AC C++ Solution:

class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
         vector<vector<int> > res;
        
        if(!pRoot)   return res;
        
        queue<TreeNode*> q;
        q.push(pRoot);
        q.push(NULL);
        vector<int> cur_vec;
        
        while(!q.empty()) {
            TreeNode *t = q.front();
            q.pop();
            
            if(t == NULL) {
                res.push_back(cur_vec);
                cur_vec.resize(0);
                if(q.size() > 0)
                    q.push(NULL);       //每层之间用一个NULL作为分隔符
            }
            else {                      //把每个节点的值存入cur_vec,并把下一层的节点加入队列
                cur_vec.push_back(t->val);  
                if(t->left) q.push(t->left);
                if(t->right) q.push(t->right);
            }
        }
        return res;
        }
    
};