leetcode算法練習【102】二叉樹的層序遍歷

所有題目源代碼：Git地址

題目

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

 

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

    3
   / \
  9  20
    /  \
   15   7
返回其層次遍歷結果：

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

方案：隊列

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
        public List<List<Integer>> levelOrder(TreeNode root) {
            // 層序
            int size = 1;
            TreeNode tmp;
            //利用隊列來實現層序遍歷
            LinkedList<TreeNode> list = new LinkedList<>(); 
            List<List<Integer>> res = new LinkedList<>();
            if(root==null) return res;
            list.add(root);
            while(list.size()>0){
                //開始遍歷
                List<Integer> nums = new LinkedList<>();
                size = list.size();
                for(int i =0;i<size;i++){
                    tmp = list.removeFirst();
                    //如果還有下一層，那就入隊
                    if(tmp.left!=null) {list.add(tmp.left);}
                    if(tmp.right!=null) {list.add(tmp.right);}
                    nums.add(tmp.val);
                }
                res.add(nums);
            }
            return res;
        }
    }

複雜度計算

• 時間複雜度：O(n)
• 空間複雜度：O(n)