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

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

   

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;
        }
    
};

©️2020 CSDN 皮肤主题: 大白 设计师: CSDN官方博客 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值