k数和

来源:互联网 发布:软件研发管理工具 编辑:程序博客网 时间:2024/04/29 08:07
给定n个不同的正整数,整数k(k < = n)以及一个目标数字。    
在这n个数里面找出K个数,使得这K个数的和等于目标数字,求问有多少种方案?
样例

给出[1,2,3,4],k=2, target=5,[1,4] and [2,3]是2个符合要求的方案

import java.util.Scanner;/** * 给定n个不同的正整数,整数k(k < = n)以及一个目标数字。    在这n个数里面找出K个数,使得这K个数的和等于目标数字,求问有多少种方案?样例给出[1,2,3,4],k=2, target=5,[1,4] and [2,3]是2个符合要求的方案 *  * @author Dell * */public class Test89 { public static int kSum(int A[], int k, int target) { int[][][] dp=new int[A.length+1][k+1][target+1];  for(int i=0;i<A.length;i++) { if(A[i]<=target) { for(int j=i+1;j<=A.length;j++) dp[j][1][A[i]]=1; }  }   for(int i=1;i<=A.length;i++) { for(int j=2;j<=k&&j<=i;j++) {for(int t=1;t<=target;t++){int x=dp[i-1][j][t];int y=0;if(t>=A[i-1]){y=dp[i-1][j-1][t-A[i-1]];}dp[i][j][t]=x+y;}  }  }   return dp[A.length][k][target];  }public static void main(String[] args) {       Scanner sc =new Scanner(System.in);       int n=sc.nextInt();       int[] a=new int[n];       for(int i=0;i<a.length;i++)       {       a[i]=sc.nextInt();       }       int k=sc.nextInt();       int target=sc.nextInt();    System.out.println(kSum(a,k,target));}}


原创粉丝点击