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

最新推荐文章于 2020-08-07 21:30:31 发布

原创最新推荐文章于 2020-08-07 21:30:31 发布 · 164 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#把二叉树打印成多行

剑指offer 专栏收录该内容

66 篇文章

订阅专栏

本文介绍了一种使用队列实现的二叉树层序遍历算法，通过记录每层节点数量，实现了按层输出的功能，每层节点从左至右依次输出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

/*
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> > vec;
            
            if(pRoot==NULL)return vec;
            
            queue<TreeNode*> q;
            q.push(pRoot);
            
            while(q.empty()!=true)
            {
                int mark_size=q.size(),i(0);
                vector<int> save;
                while(i<mark_size)
                {
                    TreeNode* ptr=q.front();
                    q.pop();
                    ++i;
                    save.push_back(ptr->val);
                    if(ptr->left!=NULL)q.push(ptr->left);
                    if(ptr->right!=NULL)q.push(ptr->right);
                }
                if(save.size()!=0)vec.push_back(save);
            }
            return vec;
        }
    
};