lintcode K數和
阿新 • • 發佈:2018-10-31
題目
給定n個不同的正整數,整數k(k < = n)以及一個目標數字。
在這n個數裡面找出K個數,使得這K個數的和等於目標數字,求問有多少種方案?
樣例
給出[1,2,3,4],k=2, target=5,[1,4] and [2,3]是2個符合要求的方案
答案
public class Solution {
/**
* @param A: An integer array
* @param k: A positive integer (k <= length(A))
* @param target: An integer
* @return : An integer
*/
public int kSum(int[] A, int k, int target) {
// write your code here
if(A == null || A.length == 0) return 0;
int[][][] DP = new int[A.length + 1][k + 1][target + 1];
DP[0][0][0] = 1;
for(int i = 1; i <= A.length; i++){
DP[i][0 ][0] = 1;
for(int j = 1; j <= k; j++)
for(int t = 1; t <= target; t++)
DP[i][j][t] = DP[i - 1][j][t] + (t >= A[i - 1] ? DP[i - 1][j - 1][t - A[i - 1]] : 0);
}
return DP[A.length][k][target];
}
}