面试题60：把二叉树打印成多行

最新推荐文章于 2021-11-29 21:19:40 发布

原创最新推荐文章于 2021-11-29 21:19:40 发布 · 320 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

52 篇文章

订阅专栏

本文介绍了一种二叉树的层次遍历算法，通过使用队列实现从上到下、从左至右的逐层节点输出。每层节点被单独成行显示，方便观察二叉树的结构。

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

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<int> vect;
            vector<vector<int> > v;
            if(pRoot == NULL)
                return v;
            queue<TreeNode *> queueNode;
            queueNode.push(pRoot);
            TreeNode *temp = NULL;
            int nextNum = 0;
            int toBePrint = 1;
            while(!queueNode.empty())
            {
                temp = queueNode.front();
                if(temp->left)
                {
                    queueNode.push(temp->left);
                    nextNum++;
                }
                if(temp->right)
                {
                    queueNode.push(temp->right);
                    nextNum++;
                }
                vect.push_back(temp->val);
                queueNode.pop();
                toBePrint--;
                if(toBePrint == 0)
                {
                    v.push_back(vect);
                    vect.resize(0);
                    toBePrint = nextNum;
                    nextNum = 0;
                }
            }
            return v;
        }
    
};