带你快速刷完67道剑指offer

No60、把二叉树打印成多行

牛客网原题链接

题目描述

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

示例1

输入

{8,6,10,5,7,9,11}

返回值

[[8],[6,10],[5,7,9,11]]

1、队列做法，保存每层的节点个数

vector<vector<int> > Print(TreeNode* pRoot) {
    vector<vector<int>> result;
    if (pRoot == nullptr) return result;
    queue<TreeNode*> q;
    q.push(pRoot);
    while (!q.empty()) {
        int len = q.size();//利用len保存每层的个数
        vector<int> temp;
        while (len--) {
            TreeNode* node = q.front();
            q.pop();
            temp.push_back(node->val);
            if (node->left)   q.push(node->left);//为空才push进去,而不是if(node) 然后直接push左右两个节点
            if (node->right)  q.push(node->right);;
        }
        result.push_back(temp);
    }
    return result;
}

二刷：

1、跟59题有点像

运行时间：2ms 占用内存：508k

vector<vector<int> > Print(TreeNode* pRoot) {
    if(pRoot == nullptr) return vector<vector<int>>();

    queue<TreeNode*> q;
    q.push(pRoot);
    vector<vector<int>> result;
    while(!q.empty()){
        int size = q.size();
        vector<int> temp;
        while(size--){
            pRoot = q.front();
            q.pop();
            temp.push_back(pRoot->val);
            if(pRoot->left)  q.push(pRoot->left);
            if(pRoot->right)  q.push(pRoot->right);

        }
        if(temp.size() > 0) result.push_back(temp);
    }
    return std::move(result);
}