lintcode K数和

题目

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