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

层序遍历利用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;
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章