给定n个不同的正整数,整数k(k < = n)以及一个目标数字。
在这n个数里面找出K个数,使得这K个数的和等于目标数字,求问有多少种方案?
样例
给出[1,2,3,4],k=2, target=5,[1,4] and [2,3]是2个符合要求的方案
分析:
动态规划dp(i , j , k)表示前i个数字中找出j个数,使得这 j 个数的和等于 k 的方案数。
则转移方程为:
dp(i, j, k) = dp(i-1, j, k) + dp(i-1, j-1, k-input[ i ])。(即取与不取第j个数的方案数之和,类似于01背包)
代码:
public class Solution {
/**
* @param A: an integer array.
* @param k: a positive integer (k <= length(A))
* @param target: a integer
* @return an integer
*/
public int kSum(int A[], int k, int target) {
// write your code here
int len = A.length;
int[][][] dp = new int[len+1][k+1][target+1];
for(int i=0;i<len;++i)
if(A[i]<=target)
for(int j=i+1;j<=len;++j)
dp[j][1][A[i]] = 1;
for(int i=1;i<=len;++i)
for(int j=2;j<=k&&j<=i;++j)
for(int s=1;s<=target;++s)
dp[i][j][s] = dp[i-1][j][s] + (s>A[i-1]?dp[i-1][j-1][s-A[i-1]]:0);
return dp[len][k][target];
}
}
写完之后,发现前 i 个数的状态 仅仅和 前 i-1 个数的状态有关,所以可以在空间上进行进一步的优化,将dp数组从三维缩小至二维。
public class Solution {
/**
* @param A: an integer array.
* @param k: a positive integer (k <= length(A))
* @param target: a integer
* @return an integer
*/
public int kSum(int A[], int k, int target) {
// write your code here
int len = A.length;
/*
int[][][] dp = new int[len+1][k+1][target+1];
for(int i=0;i<len;++i)
if(A[i]<=target)
for(int j=i+1;j<=len;++j)
dp[j][1][A[i]] = 1;
for(int i=1;i<=len;++i)
for(int j=2;j<=k&&j<=i;++j)
for(int s=1;s<=target;++s)
dp[i][j][s] = dp[i-1][j][s] + (s>A[i-1]?dp[i-1][j-1][s-A[i-1]]:0);
return dp[len][k][target];
*/
int[][] dp = new int[k+1][target+1];
dp[0][0]=1;
for(int i=1;i<=len;++i){
for(int j=k;j>=1;j--)
for(int s=target;s>=A[i-1];--s)
dp[j][s] += dp[j-1][s-A[i-1]];
}
return dp[k][target];
}
}