版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/majichen95/article/details/88925069
给定n个不同的正整数,整数k(1<= k <= n)以及一个目标数字。
在这n个数里面找出K个数,使得这K个数的和等于目标数字,你需要找出所有满足要求的方案。
样例
Example 1:
Input: [1,2,3,4], k = 2, target = 5
Output: [[1,4],[2,3]]
Example 2:
Input: [1,3,4,6], k = 3, target = 8
Output: [[1,3,4]]
解题思路:
标准的DFS解法。注意优化的点,target表示为距离当前目标还需要的数值,由于都是正整数,所以当target<0代表已经超过了,直接返回,当target==0即满足条件,当target<A[i]表示再加上A[i]也会超过,同时由于A[]是排过序,所以代表A[i]后面的元素都统统大于target,不满足条件,可直接break。
public class Solution {
/*
* @param A: an integer array
* @param k: a postive integer <= length(A)
* @param target: an integer
* @return: A list of lists of integer
*/
public List<List<Integer>> kSumII(int[] A, int k, int target) {
// write your code here
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(A);
dfs(A, k, target, new ArrayList<Integer>(), res, 0);
return res;
}
private void dfs(int[] A, int k, int target, List<Integer> list, List<List<Integer>> res, int index){
if(target < 0)
return;
if(list.size() == k && target == 0){
res.add(new ArrayList<Integer>(list));
return;
}
for(int i=index; i<A.length; i++){
if(A[i] > target)
break;
list.add(A[i]);
dfs(A, k, target - A[i], list, res, i+1);
list.remove(list.size()-1);
}
}
}