LeetCode-199-二叉樹的右檢視

雄獅虎豹發表於2022-04-19
LeetCode二叉樹

二叉樹的右檢視

題目描述：給定一個二叉樹的 根節點 root，想象自己站在它的右側，按照從頂部到底部的順序，返回從右側所能看到的節點值。

示例說明請見LeetCode官網。

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

解法一：層序遍歷二叉樹

利用佇列先進先出的特點層序遍歷二叉樹，具體處理過程如下：

  • 初始化一個佇列，並將根節點新增到佇列中，作為第一層；
  • 迴圈遍歷佇列中每一層的元素，並暫存到二維List中，並將下一層的節點按從左到右的順序加入到佇列中；
  • 直到佇列為空，遍歷結束。

最後，按順序從二維List中取每一層的最後一個元素，並返回。

import com.kaesar.leetcode.TreeNode;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

public class LeetCode_199 {
    /**
     * 層序遍歷
     * 通過使用佇列先進先出的方式遍歷二叉樹
     *
     * @param root
     * @return
     */
    public static List<Integer> rightSideView(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        // 暫存每一層的節點
        List<List<Integer>> numsList = new ArrayList<>();
        Queue<TreeNode> nodes = new LinkedList<>();
        nodes.add(root);
        // 遍歷每一層的節點，並將下一層的節點按從左到右的順序加入到佇列中
        while (!nodes.isEmpty()) {
            List<Integer> nums = new ArrayList<>();
            int count = nodes.size();
            for (int i = 0; i < count; i++) {
                TreeNode cur = nodes.poll();
                nums.add(cur.val);
                if (cur.left != null) {
                    nodes.add(cur.left);
                }
                if (cur.right != null) {
                    nodes.add(cur.right);
                }
            }
            numsList.add(nums);
        }

        List<Integer> result = new ArrayList<>();
        // 按順序取每一層的最後一個元素
        for (List<Integer> integers : numsList) {
            result.add(integers.get(integers.size() - 1));
        }
        return result;
    }

    public static void main(String[] args) {
        /**
         * 測試用例：
         *     1
         *   2   3
         *    5   4
         */
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.right = new TreeNode(3);
        root.left.right = new TreeNode(5);
        root.right.right = new TreeNode(4);

        // 期望輸出： 1,3,4
        for (Integer integer : rightSideView(root)) {
            System.out.print(integer + " ");
        }
    }
}
【每日寄語】 如果一個人努力尋求向上，做好自己的工作，為自己尋找幸福，是好勝的話那麼我就好勝！

相關文章