层序遍历二叉树(队列的使用)

层序遍历利用BFS搜索算法,利用queue,写了c++和Java两个版的。


c++中队列queue< T >的pop()方法返回void类型而非T类型,要先用front()获取

(c++)代码如下:

    vector<int> PrintFromTopToBottom(TreeNode* root) {
           queue<TreeNode *> q;
           vector<int> rst;
        if(root==NULL){
            return rst;
        }
        q.push(root);
        while(!q.empty()){
            int size=q.size();
            for(int i=0;i<size;i++){
                TreeNode *cur=q.front();
                q.pop();
                rst.push_back(cur->val);
                if(cur->left!=NULL){
                    q.push(cur->left);
                }
                if(cur->right!=NULL){
                    q.push(cur->right);
                }
            }
        }
        return rst;
    }

java中队列Queue的方法简述:

add 增加一个元索,如果队列已满,则抛出一个IIIegaISlabEepeplian异常
remove 移除并返回队列头部的元素,如果队列为空,则抛出一个NoSuchElementException异常
element 返回队列头部的元素, 如果队列为空,则抛出一个NoSuchElementException异常
offer 添加一个元素并返回true,如果队列已满,则返回false
poll 移除并返问队列头部的元素, 如果队列为空,则返回null
peek 返回队列头部的元素, 如果队列为空,则返回null
put 添加一个元素,如果队列满,则阻塞
take 移除并返回队列头部的元素,如果队列为空,则阻塞

(Java)代码如下:


 public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> rsts=new ArrayList<>();
        if(root==null)
            return rsts;
        Queue<TreeNode> q=new LinkedList<TreeNode>();
        q.offer(root);
        while(!q.isEmpty()){
            ArrayList<Integer> rst=new ArrayList<>();
            int size=q.size();
            for(int i=0;i<size;i++){
                TreeNode head=q.poll();
                rst.add(head.val);
                if(head.left!=null) 
                    q.offer(head.left);
                if(head.right!=null)
                    q.offer(head.right);
            }
            rsts.add(rst);
        }
        return rsts;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值