1. 程式人生 > >lintcode K數和

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