LeetCode-107-二叉樹的層序遍歷 II

雄獅虎豹發表於2021-11-21
LeetCode二叉樹

二叉樹的層序遍歷 II

題目描述：給定一個二叉樹，返回其節點值自底向上的層序遍歷。 （即按從葉子節點所在層到根節點所在的層，逐層從左向右遍歷）

示例說明請見LeetCode官網。

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/probl...
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

解法一：層序遍歷
  • 首先，如果根節點為空，直接返回空的結果集。

  • 如果根節點不為空，通過佇列來遍歷每一層的節點，具體處理過程如下：

    • 首先將根節點放入佇列；
    • 遍歷佇列中當前的節點數，即為當前層的結果，然後再將當前層節點的左右非空子節點放入到佇列中 ；
    • 然後繼續遍歷佇列中下一層的節點，直到佇列為空位置。
  • 這樣得到的結果是從上往下層序遍歷的結果， 最後呼叫Collections.reverse(result);這個方法，將得到的結果集逆序排列，即可得到自底向上的層序遍歷。
import com.kaesar.leetcode.TreeNode;

import java.util.*;

public class LeetCode_107 {
    /**
     * 層序遍歷
     *
     * @param root
     * @return
     */
    public static List<List<Integer>> levelOrderBottom(TreeNode root) {
        // 初始化結果集
        List<List<Integer>> result = new ArrayList<>();
        // 如果根節點為空，直接返回result
        if (root == null) {
            return result;
        }
        // 用一個佇列存每一層的節點
        Queue<TreeNode> nodes = new LinkedList<>();
        nodes.add(root);
        // 遍歷每一層的節點，將節點的值放入list中
        while (!nodes.isEmpty()) {
            int count = nodes.size();
            List<Integer> curLevel = new ArrayList<>();
            while (count > 0) {
                TreeNode cur = nodes.poll();
                curLevel.add(cur.val);
                if (cur.left != null) {
                    nodes.add(cur.left);
                }
                if (cur.right != null) {
                    nodes.add(cur.right);
                }
                count--;
            }
            result.add(curLevel);
        }
        // 最後將得到的結果集逆序，得到從最下層 -> 最上層的遍歷結果
        Collections.reverse(result);
        return result;
    }

    public static void main(String[] args) {
        // 測試用例
        TreeNode root = new TreeNode(3);
        root.left = new TreeNode(9);
        root.right = new TreeNode(20);
        root.right.left = new TreeNode(15);
        root.right.right = new TreeNode(7);

        for (List<Integer> integers : levelOrderBottom(root)) {
            for (Integer integer : integers) {
                System.out.print(integer + " ");
            }
            System.out.println();
        }
    }
}
【每日寄語】 多少事，從來急，天地轉，光陰迫，一萬年太久，只爭朝夕。

相關文章