vijos 1128 N个数选K个数 (DFS )

从 n 个整数中任选 k 个整数相加，可分别得到一系列的和 要求你计算出和为素数共有多少种

IN
4 3
3 7 12 19

OUT
1

# include <iostream>
# include <cstdio>
# include <cmath>
using namespace std ;

int a[22] ;
int ans = 0 ;
int n , k ;

bool isp (int val )  
{
     int i ;
     if (val == 1)
       return 0 ;
     for (i = 2 ; i <= sqrt(val) ; i++)
     {     
         if (val % i == 0)
             return 0 ;
     }
     return 1 ;

}

void dfs(int cur , int cnt , int num)
{
    if (cnt == k)
    {
        if (isp(num))
           ans++ ;
        return ;
    }
    int i ;
    for (i = cur ; i <= n ; i++)
       dfs(i+1 , cnt+1 , num+a[i]) ;
}

int main ()
{

    while (scanf("%d %d" ,&n , &k) != EOF)
    {
        ans = 0 ;
        int i ;
        for (i = 1 ; i <= n ; i++)
            scanf("%d" , &a[i]) ;
        dfs(1 , 0 , 0) ;
        printf("%d
" , ans) ;
    }
    
    
    return 0 ;
}