力扣面试题 04.12. 求和路径

最新推荐文章于 2024-10-04 14:10:29 发布

eyvr

最新推荐文章于 2024-10-04 14:10:29 发布

阅读量65

点赞数

文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/qq_46636391/article/details/125331434

版权

在这里插入图片描述
O(n²)的做法，dfs遍历整棵树，遍历的过程中查找是否有sum的和

class Solution {
    int ans = 0;
    public int pathSum(TreeNode root, int sum) {
        find(root, sum, new ArrayList<Integer>());
        return ans;
    }
    void find(TreeNode t, int sum, List<Integer> l){
        if(t == null){
            return;
        }
        List<Integer> li = new ArrayList<>();
        for(int i = 0; i < l.size(); i++){
            li.add(l.get(i) + t.val);
            if(li.get(i) == sum)
                ans++;
        }
        li.add(t.val);
        if(t.val == sum)
            ans ++;
        find(t.left, sum, li);
        find(t.right, sum, li);
    }
}