樹的層序遍歷(深度優先 和 隊列)

題目: Given a binary tree, return the bottom-up level order traversal of its nodes’ values. (ie, from left to right, level by level from leaf to root).

DFS算法:

public List<List<Integer>> levelOrderBottom(TreeNode root) {
    List<List<Integer>> res = new ArrayList<List<Integer>>();
    helper(root,0,res);
    return res;
}
// 一個前序遍歷 
public void helper(TreeNode root, int n, List<List<Integer>> res){
    if(root==null) return;
    // 取得同一層的鏈表 如果沒有則 添加一個
    if(res.size()<=n) res.add(0,new ArrayList<Integer>());
    res.get(res.size()-n-1).add(root.val);
    helper(root.left,n+1,res);
    helper(root.right,n+1,res);
}

隊列實現:

/**
 * 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>> levelOrderBottom(TreeNode root) {
        LinkedList<List<Integer>> result = new LinkedList<> ();
        LinkedList<TreeNode> queue = new LinkedList<> ();

        if(root == null) return result;
        int cur = 1;
        int next = 0;
        queue.add(root);
        List<Integer> list = new ArrayList<> ();

        int count = 0;
        while(!queue.isEmpty()) {
            // 遍歷完一層
            if(count == cur) {
                cur = next;
                result.addFirst(list);
                list = new ArrayList<> ();
                // 這兩個置0
                count = 0;
                next = 0;
            }
            TreeNode node = queue.removeFirst();
            list.add(node.val);
            count++;
            if(node.left != null) {
                next++;
                queue.add(node.left);
            }
            if(node.right != null) {
                next++;
                queue.add(node.right);
            }
            if(queue.isEmpty()) {
                result.addFirst(list);
            }
        }

        return result;
    }

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