javaP1036 [NOIP2002 普及组] 选数

 

题目描述

已知 �n 个整数 �1,�2,⋯ ,��x1​,x2​,⋯,xn​，以及 11 个整数 �k（�<�k<n）。从 �n 个整数中任选 �k 个整数相加，可分别得到一系列的和。例如当 �=4n=4，�=3k=3，44 个整数分别为 3,7,12,193,7,12,19 时，可得全部的组合与它们的和为：

3+7+12=223+7+12=22

3+7+19=293+7+19=29

7+12+19=387+12+19=38

3+12+19=343+12+19=34

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3+7+19=293+7+19=29。

输入格式

第一行两个空格隔开的整数 �,�n,k（1≤�≤201≤n≤20，�<�k<n）。

第二行 �n 个整数，分别为 �1,�2,⋯ ,��x1​,x2​,⋯,xn​（1≤��≤5×1061≤xi​≤5×106）。

输出格式

输出一个整数，表示种类数。

输入输出样例

    import java.io.BufferedReader;
    import java.io.IOException;
    import java.io.InputStreamReader;
    import java.io.StreamTokenizer;
    import java.math.BigDecimal;
    import java.math.BigInteger;
    import java.util.*;

    public class Main {
        private static int []a;
        private static int []book;
        private static int k;
        private static int count;
        private static int n;
        public static void main(String[] args) {
            Scanner sc = new Scanner(System.in);

            n=sc.nextInt();
            k=sc.nextInt();
            count=0;
            a=new int[n];
            book=new int[n];
            for(int i=0;i<n;i++){
                a[i]=(sc.nextInt());
            }
            dfs(0,0,0);
            System.out.println(count);
        }
        public static void dfs(int depth,int sum,int start){
            boolean flag=true;
            if(depth==k){
                for(int i=2;i<sum;i++){
                    if(sum%i==0){
                        flag=false;
                    }
                }

                if(flag){
                    count++;
                }
                return;
            }
            for(int i=start;i<n;i++){
             //   if(book[i]==0){
               //     book[i]=1;
                    dfs(depth+1,sum+a[i],i+1);
                    //book[i]=0;
              //  }
            }

        }

    }