洛谷 P1036 选数 (简单递归)

题目描述

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

3＋7＋12=22

3＋7＋19＝29

7＋12＋19＝38

3＋12＋19＝34。

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

例如上例，只有一种的和为素数：3＋7＋19＝29）。

输入输出格式

输入格式：

键盘输入，格式为：

n , k （1<=n<=20，k＜n）

x1,x2，…,xn （1<=xi<=5000000）

输出格式：

屏幕输出，格式为：

一个整数（满足条件的种数）。

代码：

#include<stdio.h>
#include<math.h>

int ans=0,sum=0,n,k,v[25];

void judge()
{
    if(sum==1)return;
    int s=sqrt(sum)+1,i;
    for(i=2;i<s;i++)
        if(!(sum%i))break;
    if(i==s)ans++;
}

void dfs(int deep,int beg)
{
    if(deep==k)
    {
        judge();
        return;
    }
    for(int i=beg;i<n;i++)
    {
        sum+=v[i];
        dfs(deep+1,i+1);
        sum-=v[i];
    }
}

int main()
{
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++)scanf("%d",&v[i]);
    dfs(0,0);
    return !printf("%d\n",ans);
}