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;
    }
}


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