洛谷 P1036 选数（dfs）

洛谷 P1036 选数

原题链接

dfs深度搜索

通过阅读题目，可知要求和为素数的个数，其中，在n个数中选出k个数然后求和，判断素数个数。我主要把这道题分为三个步骤：
1、组合数
2、求和
3、判断素数
其中，对于2、3，大家应该都非常熟悉，尤其是素数判断，最基础的知识了，难点就是在怎么来组合数这个问题上，大家是不是想到了“全排列”，哈哈哈哈，就是“全排列”，一种小变形吧，不同的是，我们要的是“全组合”，并非‘排列’，在数中选数，得到不同的和，进而判断素数个数。而在这个组合中，就是去重了。
举个例子：
比如说在6里面随便选5个数，那么选法都是什么呢？
瞎枚举？
12345
12346
为防止重复，可保证枚举的这些数是升序排列（或者理解为高数里面的“单调不减函数”），可以是平的,比如 1 2 2 3 3 4…
但是这道题不能平，否则就有重复数字了。

6个数选3个数-举例子
1 2 3
1 2 4
1 2 5
1 2 6
第一轮枚举完毕。
第二个数加一
1 3 ？
这个“？”应该是4，因为是升序排列
1 3 4
1 3 5
1 3 6
接着，就是这样
1 4 5
1 4 6
1 5 6
第一位是1枚举完毕
第一位是2呢?
2 3 4
2 3 5
2 3 6
2 4 5
2 4 6
2 5 6
就是这样的，枚举还是蛮清晰的吧
以此类推…
3 4 5
3 4 6
3 5 6
4 5 6
然后就枚举不了了，结束。
所以说，这样就可以避免判重了。

然后，就开始深搜，根据上面的模板，就可以写出来了

 #include <bits/stdc++.h>
 #include <iostream>

 using namespace std;
 #define N 21

 int a[N],sum=0,ans=0;//ans符合条件个数
 int n,k;//全局变量

 bool isprime(int a)//素数判断
 {
     if(a==2)
        return true;
     for(int i=2;i<=sqrt(a);i++)
        if(a%i==0)
          return false;
     return true;
 }
 void dfs(int x,int y)//深搜
 //x:当前正在选择第x个数，其和用sum储存
 //y：当前最后选过的数在数组中的位置
 {
     if(x>k)//若选的数比k多，就检测isprime
     {
         if(isprime(sum))
            ans++;
         return;
     }//判断条件，符合.
     for(int i=y+1;i<=n;i++)//选数不足k，就继续搜索下一个
     {
         sum+=a[i];//取数->求和
         dfs(x+1,i);//搜索
         sum-=a[i];//回溯
     }
 }

 int main()
 {
     cin>>n>>k;
     for(int i=1;i<=n;i++)
        cin>>a[i];
     sort(a+1,a+n);//不过发现有没有它不影响结果...
     dfs(1,0);//选择第一个数，因还未选过数，y=0
     cout<<ans<<"\0";
 }

不当之处，还请指出~