層序遍歷二叉樹(隊列的使用)

層序遍歷利用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;
    }