LeetCode-113-路徑總和 II

雄獅虎豹 發表於 2021-11-25
LeetCode

路徑總和 II

題目描述:給你二叉樹的根節點 root 和一個整數目標和 targetSum ,找出所有 從根節點到葉子節點 路徑總和等於給定目標和的路徑。

葉子節點 是指沒有子節點的節點。

示例說明請見LeetCode官網。

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

解法一:層序遍歷

Queue<Pair<TreeNode, Pair<Integer, List<Integer>>>>這種結構來記錄當前節點的路徑以及路徑和,其中:

  • 外層的Pair的key為當前節點;
  • 內層的Pair的key為當前路徑的總和,value為當前路徑的記錄,從根節點到當前節點。

然後使用層序遍歷的方式使用佇列遍歷二叉樹的節點,當判斷某節點的左右節點為空,即為葉子節點,然後判斷當前的路徑值是否與targetSum相等,如果相等,將相應的路徑新增到結果集中。

最後,返回結果集。

import com.kaesar.leetcode.TreeNode;
import javafx.util.Pair;

import java.util.*;

public class LeetCode_113 {
    // 結果集
    private static List<List<Integer>> result = new ArrayList<>();

    public static List<List<Integer>> pathSum(TreeNode root, int targetSum) {
        if (root == null) {
            return new ArrayList<>();
        }
        /**
         * 外層的Pair的key為當前節點
         * 內層的Pair的key為當前路徑的總和,value為當前路徑的記錄,從根節點到當前節點
         */
        Queue<Pair<TreeNode, Pair<Integer, List<Integer>>>> nodes = new LinkedList<>();
        List<Integer> path = new ArrayList<>();
        path.add(root.val);
        // 初始化,將根節點新增到佇列中
        nodes.add(new Pair<>(root, new Pair<>(root.val, path)));

//        while (!nodes.isEmpty()) {
            Pair<TreeNode, Pair<Integer, List<Integer>>> cur = nodes.poll();
            TreeNode curNode = cur.getKey();
            // 判斷當前節點的左右節點為空,即為葉子節點,然後判斷當前的路徑值是否與targetSum相等
            if (curNode.left == null && curNode.right == null) {
                if (cur.getValue().getKey() == targetSum) {
                    result.add(cur.getValue().getValue());
                }
                continue;
            }
            // 如果當前節點不是葉子節點,繼續往下遍歷
            if (curNode.left != null) {
                List<Integer> leftPath = new ArrayList<>(Arrays.asList(new Integer[cur.getValue().getValue().size()]));
                Collections.copy(leftPath, cur.getValue().getValue());
                leftPath.add(curNode.left.val);
                nodes.add(new Pair<>(curNode.left, new Pair<>(cur.getValue().getKey() + curNode.left.val, leftPath)));
            }
            if (curNode.right != null) {
                List<Integer> rightPath = new ArrayList<>(Arrays.asList(new Integer[cur.getValue().getValue().size()]));
                Collections.copy(rightPath, cur.getValue().getValue());
                rightPath.add(curNode.right.val);
                nodes.add(new Pair<>(curNode.right, new Pair<>(cur.getValue().getKey() + curNode.right.val, rightPath)));
            }
        }

        return result;
    }

    public static void main(String[] args) {
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.right = new TreeNode(3);
        root.right.left = new TreeNode(4);
        root.right.right = new TreeNode(5);

        for (List<Integer> integers : pathSum(root, 8)) {
            for (Integer integer : integers) {
                System.out.print(integer + " ");
            }
            System.out.println();
        }
    }
}
【每日寄語】 命運,不過是失敗者無聊的自慰,不過是懦怯者的解嘲。人們的前途只能靠自己的意志、自己的努力來決定。