洛谷 P1036 选数

看了题解，竟然有大佬枚举出来了，最多用了10个for，tql%%%

刚开始就想着用深搜了，第一发wa了，重复计算了，后来改过来了

题目

已知 n 个整数 x₁,x₂,…,x_n，以及11个整数kk(k<nk<n)。从n个整数中任选k个整数相加，可分别得到一系列的和。例如当n=4,k=3,4个整数分别为3,7,12,19时，可得全部的组合与它们的和为：

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

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

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

3+12+19=34  3+12+19=34。

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

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

输入输出格式

输入格式：

键盘输入，格式为：

n , k ( 1 ≤ n ≤ 20 , k < n )

x₁ , x₂ , … , x_n ( 1 ≤ x_i ​≤ 5000000)

输出格式：

屏幕输出，格式为： 1个整数（满足条件的种数）。

输入输出样例

输入样例#1： 

4 3
3 7 12 19

 

输出样例#1： 

1

题解

// luogu-judger-enable-o2
#include <iostream>
#include <cstdio>
#include <cmath>

using namespace std;
void dfs(int num);   //深搜
bool check(int num); //判断这个数在前面出现过没有
bool judge(int num); //判断合法的和是不是素数
bool fun(int num);   //避免重复计算，保证后面的数下标比前面的数下标大
int n, k, ans, a[30], b[30];

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

void dfs(int num)
{
    if(num >= k)
    {
        if(fun(num))
        {
            int temp = 0;
            for(int i = 0; i < k; i ++ )
            temp += a[b[i]];
            //printf("%d\n",temp);
            if(judge(temp))  ans ++ ;
            return;     
        } else  return;
    }
    for(int i = 0; i < n; i ++ )
    {
        b[num] = i;
        if(check(num))
            dfs(num + 1);
    }
}

bool check(int num)
{
    for(int i = 0; i < num; i ++ )
        if(b[i] == b[num])  return 0;
    return 1;
}

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

bool fun(int num)
{
    for(int i = 0; i < num - 1; i ++ )
      if(b[i] > b[i + 1])  return 0;
    return 1;
}