【算法/剑指Offer】从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。

题目描述

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


/*
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> > vv;
            vector<int> ve;
            queue<TreeNode*> que;
            queue<int> dque;
            int depth=0;
            que.push(pRoot);
            dque.push(depth);
            TreeNode *node;
            while(!que.empty())
            {
                node=que.front();
                if(depth!=dque.front())
                {
                    vv.push_back(ve);
                    ve.clear();
                    depth=dque.front();
                }
                que.pop();
                dque.pop();
                if(node)
                {
                    que.push(node->left);
                    que.push(node->right);
                    dque.push(depth+1);
                    dque.push(depth+1);
                    ve.push_back(node->val);
                }
            }
            return vv;
        }
};

链接:https://www.nowcoder.com/questionTerminal/445c44d982d04483b04a54f298796288
来源:牛客网

//用两个队列实现按层打印树
    vector<vector<int> > Print2(TreeNode* pRoot)
    {
        vector<vector<int> > res;
        if(pRoot==nullptr) return res;
        queue<TreeNode*> q1,q2;//用队列缓存树节点的print顺序
        q1.push(pRoot);
        while(!q1.empty())
        {
            vector<int> tmp;
            while(!q1.empty())
            {
                tmp.push_back(q1.front()->val);//依次取出栈中元素的值
                if(q1.front()->left) q2.push(q1.front()->left);//将栈顶元素按从左至右顺序压入队列
                if(q1.front()->right) q2.push(q1.front()->right);
                q1.pop();//删除栈顶元素
            }
            while(!q2.empty()) //将队列中的元素依次入栈
            {
                q1.push(q2.front());
                q2.pop();
            }
            res.push_back(tmp);
        }
        return res;
    }

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值