c++ 组合and判断素数

题目描述

已知 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）

第二行n个整数：x1,x2，…,xn （1<=xi<=5000000）

输出

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

样例输入

4 3
3 7 12 19

样例输出

1

Source Code

#include <iostream>
#include <cmath>
#include <string.h>
#include <stdio.h>
using namespace std;
int n,m,a[100],s[100],sum,ans;
bool isPrime(int);
void dfs(int x,int y)
{
    if(x == m)//如果挑选出来3组数据,就求和,小于3个数,就继续挑选
    {
        sum = 0;
        for(int i = 1;i <= x;i ++)
        {
            printf("a[%d] = %dn",i,a[i]);
            sum += s[a[i]];//对挑出来的三个数t求和
        }
        cout << "------------" << endl;
        if(isPrime(sum) == 1)
            ans ++;//如果是和是素数,就+1
        return ;
    }
    y ++;
    for(int i = y;i <= n;i ++)//从4个数中挑选三个数字
    {
        a[x + 1] = i;//改变下标形成新的挑选数
        dfs(x + 1,i);//当x + 1等于3 就开始求和  如果小于3就继续挑选
    }
}
bool isPrime(int r)//判断是否是素数
{
    for(int i = 2;i <= sqrt(r);i ++)
        if(r % i == 0)
            return 0;
    return 1;
}
int main()
{
    cin >> n >> m;
    for(int i = 1;i <= n;i ++)
        cin >> s[i];
    dfs(0,0);
    cout << ans << endl;
    return 0;
}