【两次过】Lintcode 90. k数和 II

版权声明：本文为博主原创文章，未经博主允许不得转载。 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);
        }
    }
}