(Java)leetcode-102 Binary Tree Level Order Traversal (二叉樹的層序遍歷)

題目

給你一個二叉樹,請你返回其按 層序遍歷 得到的節點值。 (即逐層地,從左到右訪問所有節點)。

示例:
二叉樹:[3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其層次遍歷結果:

[
[3],
[9,20],
[15,7]
]

思路

參考leetcode-429 N-ary Tree Level Order Traversal(N叉樹的層序遍歷)
用隊列來實現。節點出隊則將其孩子入隊,用for循環來保證將某一層的節點全部出隊,此時表明同一層的節點都處理完畢,將此層節點值添加到結果集合中。

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new LinkedList<>();
        if (root == null) return res;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()) {
        	List<Integer> levelRes = new LinkedList<>();
        	for (int i = queue.size(); i > 0; i--) {
        		TreeNode current = queue.poll();
        		levelRes.add(current.val);
        		if (current.left != null) queue.offer(current.left);
        		if (current.right != null) queue.offer(current.right);
        	}
        	res.add(levelRes);
        }
        return res;
    }
}

執行用時:1 ms, 在所有 Java 提交中擊敗了90.67%的用戶
內存消耗:39.9 MB, 在所有 Java 提交中擊敗了5.71%的用戶

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