LeetCode102: Binary Tree Level Order Traversal

題目要求:https://leetcode.com/problems/binary-tree-level-order-traversal/

1)此題本質上是一個BFS問題,不難解決

2)在對queue遍歷時需要注意
  • 題目要求將每一層節點作爲一個list,那麼使用while(!q.isEmpty())的方式來遍歷Queue就不太合適,因爲需要不停地將節點加入隊列。
  • 這裏在訪問每一層時,先獲得此時queue裏的長度,用for循環遍歷。
  • 使用Iterator()同樣可行

AC代碼:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        List<List<Integer>> resList = new LinkedList<List<Integer>>();
        if(root == null) return resList;
        q.add(root);
        
        while(!q.isEmpty()){
            List<Integer> levelList = new LinkedList<Integer>();
            int qSize = q.size();
            for(int i=0; i<qSize; i++){
                TreeNode tmp = q.poll();
                levelList.add(tmp.val);
                if(tmp.left != null){
                    q.add(tmp.left);
                }
                if(tmp.right != null){
                    q.add(tmp.right);
                }
            }
            resList.add(levelList);
        }
        return resList;
    }
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章