欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode -- 39. 组合总数 java

程序员文章站 2024-03-15 09:13:23
...

LeetCode -- 39. 组合总数 java
LeetCode -- 39. 组合总数 java
LeetCode -- 39. 组合总数 java
来自图解面试算法, 动画来自吴师兄。
要求得出所有和为目标值的组合,那么这样使用回溯法将所有结果遍历一边是非常合适的。
数组中所有元素都是可以重复使用的,那么也就是说每次都可以从第一位开始,但是最终结果中不允许包括重复的组合,那么应该怎样解决呢,那就要将每组数据的第一个值的索引传递进去,每一次都是从这个位置开始判断,那么就可以避免将之前的元素重复使用了。

	List<List<Integer>> result = new ArrayList<>();  // 最终结果
	
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        Arrays.sort(candidates);
        for (int i = 0; i < candidates.length; i++) {
            ArrayList<Integer> resultList= new ArrayList<>();  // 其中一个可能的结果
            resultList.add(candidates[i]);
            dfs(candidates, target, candidates[i], resultList, i);
        }
        return result;
    }
    
    private void dfs(int[] candidates, int target, int val, List<Integer> r, int j){  // val是candidates[i], r是resultList,j是i
        if (target==val){
            result.add(new ArrayList<>(r));
            return;  
        }
        if (val>target){
            return;  
        }
        for (int i = j; i < candidates.length; i++) {  // 从当前位置开始循环
            r.add(candidates[i]);  // r先添加了当前的数
            val+=candidates[i];  // val代表当前数组总和,更新val
            dfs(candidates,target,val,r,i);  // 用更新的val和i进行递归
            //回溯
            r.remove(r.size() - 1);  // remove的参数可以是index,也可以是值,这里r.size-1的类型是int,不是Integer,即索引
            // 即删除最后一个数字,因为加过了
            val-=candidates[i];
        }
    }