leetcode *102. 二叉樹的層序遍歷

【題目】*102. 二叉樹的層序遍歷

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

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

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

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

【解題思路1】BFS - 隊列

List<List> ans = new ArrayList<>();
Queue queue = new LinkedList<>();
List list = new ArrayList<>();

/**
 * 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) {
        List<List<Integer>> ans = new ArrayList<>();
        if(root == null){
            return ans;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            int size = queue.size();
            List<Integer> list = new ArrayList<>();
            while(size > 0){
                TreeNode cur = queue.poll();
                list.add(cur.val);
                size--;
                if(cur.left != null){
                    queue.offer(cur.left);
                }
                if(cur.right != null){
                    queue.offer(cur.right);
                }
            }
            ans.add(list);
        }
        return ans;
    }
}

【解題思路2】DFS 遞歸

相同層的放入同一個數組

class Solution {
    private int num = 0;//標識現在遍歷的層次
    private List<List<Integer>> results = new LinkedList<List<Integer>>();
    public List<List<Integer>> levelOrder(TreeNode root) {
        //null處理
        if(root == null) {
            return results;
        }
        //深度遍歷
        DFS(root);
		return results;
    }
   void DFS(TreeNode node) {
       //構造result保存每一層的數據
       if(results.size() <= num) {
           results.add(new LinkedList<Integer>());
       }
        //add數據到result
       results.get(num).add(node.val);
        //遍歷左子樹
       if(node.left != null) {
           //num+1表示遍歷深了一層
           num++;
           DFS(node.left);
            //遍歷完成返回後, 層次應該相應減少一層
           num--;
       }
        //遍歷右子樹, 與前面一樣
       if(node.right != null) {
           num++;
           DFS(node.right);
           num--;
       }
       return;

   }
}