【Lintcode】90. k Sum II

题目地址：

https://www.lintcode.com/problem/k-sum-ii/description

给定一个长度为$n$的正整数数组，数组内数字彼此不同，再给定两个数$k$和$t$，要求在数组中找所有的$k$个数的组合其和为$t$。

思路是DFS。为了剪枝方便，可以先排个序。剪枝的技巧在于：
1、如果枚举时发现剩下来要枚举的数不足以填满$k$个数了，就直接退出本层枚举（因为接下来的数就算全加进去，也不够$k$个），回到递归上一层；
2、如果发现某次已经枚举的数已经到达了$k$个，但和仍然不为$t$，也可以直接退出本层枚举（因为继续枚举会得到size大于$k$的组合），回到递归上一层；
3、如果发现枚举到一个大于$t$的数，也可以退出本层枚举，回到递归上一层（因为接下来枚举会得到更大的和，不会得到合法的解，所以可以直接退出）。

代码如下：

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

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<>();
        if (A == null || A.length == 0) {
            return res;
        }
    
        Arrays.sort(A);
        
        dfs(A, k, target, 0, new ArrayList<>(), res);
        return res;
    }
    
    private void dfs(int[] A, int k, int target, int pos, List<Integer> cur, List<List<Integer>> res) {
        if (target == 0 && cur.size() == k) {
            res.add(new ArrayList<>(cur));
            return;
        }
    	
    	// 这里是考虑的剪枝情况2
		if (cur.size() == k) {
            return;
        }
		
		// 这里k - cur.size() <= A.length - i考虑的是剪枝情况1
        for (int i = pos; k - cur.size() <= A.length - i; i++) {      
            if (target >= A[i]) {
                cur.add(A[i]);
                dfs(A, k, target - A[i], i + 1, cur, res);
                cur.remove(cur.size() - 1);
            } else {
            	// 这里是考虑的剪枝情况3
                return;
            }
        }
    }
}

时间复杂度$O(k(\genfrac{}{}{0ex}{}{n}{k}))$，空间复杂度$O(k)$。