LeetCode113【DFS】：路径总和2

最新推荐文章于 2024-07-03 22:53:03 发布

Andy Shaw

最新推荐文章于 2024-07-03 22:53:03 发布

阅读量189

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/baidu_34209307/article/details/106535482

版权

LeetCode 专栏收录该内容

27 篇文章 0 订阅

订阅专栏

113. 路径总和 II

难度中等

给定一个二叉树和一个目标和，找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树，以及目标和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

[
   [5,4,11,2],
   [5,8,4,5]
]

    class Solution {
        List<List<Integer>> ans = new ArrayList<>();

        public List<List<Integer>> pathSum(TreeNode root, int sum) {
            if (root == null) {
                return ans;
            }
            List<Integer> inputList = new ArrayList<>();
            inputList.add(root.val);
            dfs(root, inputList, sum, root.val);
            return ans;
        }

        public void dfs(TreeNode root, List<Integer> tempArray, int sum, int sumNow) {
            if (root.left == null && root.right == null && sumNow == sum) {
                ans.add(new ArrayList<>(tempArray));
            }

            if (root.left != null) {
                List<Integer> tempList = new ArrayList<>(tempArray);
                tempList.add(root.left.val);
                dfs(root.left, tempList, sum, sumNow + root.left.val);
            }
            if (root.right != null) {
                List<Integer> tempList = new ArrayList<>(tempArray);
                tempList.add(root.right.val);
                dfs(root.right, tempList, sum, sumNow + root.right.val);
            }
        }
    }