k数和

给定n个不同的正整数，整数k（k < = n）以及一个目标数字。　　　　

在这n个数里面找出K个数，使得这K个数的和等于目标数字，求问有多少种方案？

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出[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];
    }
}