P1036 选数（DFS）

题目链接：https://www.luogu.org/problemnew/show/P1036

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

输入输出样例
输入样例#1：
4 3
3 7 12 19
输出样例#1：
1

困惑了我好久的问题：为什么一个需要减掉当前所加的，一个不需要？分别对应解法 1、2 和 3、4；
说明一个问题： 就是定义成全局变量存储和时，以及定义成参数存储变量时的区别；

1、定义成全局变量num时，在执行for循环一次dfs结束时，要想走下一次for循环寻找情况时，必须将当前加的a[j]减掉，否则会造成同一层既加了a[j],又加了下一次for循环的a[j]。

2、定义成参数时，在执行for循环一次dfs结束时，此时的参数为sum+a[j]，他执行结束时，没有减掉a[j],是因为在进行下一次for循环时，传的参数仍仍然是sum+a[j]，关键点：此时的sum和上一次的sum是一样的，因为这两个dfs，属于同一层，传来的参数都是上一次dfs传来的sum，若是减掉a[j]，会造成减多了，计算不正确。。

开始发布题解：就是dfs搜索所有情况，计数器 i 若到达边界k时进行判断是不是质数，并return回去。

关键点：这样相当于全排列，程序会把所有情况都找出来，如：1、2、3 和 3、1、2，本来应该计数一次，现在这种两种情况都进行了计数。但是我们可以将最后算出来的结果除以k的阶乘就行，可以去掉重复的，就是排列组合的排列。显然：这样会增加不少运算量，所以有了解法二。

第一次：

#include <iostream>
#include <cstdio>
#include <string>
#include <string.h>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
typedef long long ll;
#define maxn 10005
#define mod 7654321

int num,n,k,numm;
int a[100];
bool vis[100];

int f(int b)
{
    if(b==1||b==0)
    {
        return 0;
    }
    for(int i=2;i*i<=b;i++)
    {
        if(b%i==0)
            return 0;
    }

    return 1;

}

void dfs(int i)//k=3
{
    if(i>=k)//i从0开始，0,1,2,3
    {
        if(f(num))
            numm++;//cout<<num<<endl;
        return;
    }
    else
    {
        for(int j=0;j<n;j++)
        {
            if(!vis[j])
            {
                num+=a[j];
                vis[j]=true;

                dfs(i+1);

                vis[j]=false;
                num-=a[j];

            }
        }
    }
}

int main()
{
    cin>>n>>k;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
    }
    dfs(0);
    int sum=1;
    for(int i=1;i<=k;i++)
    {
        sum*=i;
    }

    cout<<numm/sum<<endl;

    return 0;
}

优化方法（去重）： 用一个start参数来判断每次for循环走到的位置，下一次循环只需要从上一次的位置往后找就行了。

原因：只需要保证下标从小到大就可以去重。不解释了。

第二次优化：

#include <iostream>
#include <cstdio>
#include <string>
#include <string.h>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
typedef long long ll;
#define maxn 10005
#define mod 7654321

int num,n,k,numm;
int a[100];
bool vis[100];

bool isprime(int b)//判断质数
{
    if(b<=1)
    {
        return false;
    }
    for(int i=2;i*i<=b;i++)
    {
        if(b%i==0)
            return false;
    }
    return true;
}

void dfs(int i,int start)
{
    if(i>=k)
    {
        if(isprime(num))
            numm++;
        return;
    }
    for(int j=start;j<n;j++)//关键去重，另 j=start
    {
        if(!vis[j])
        {
            num+=a[j];
            vis[j]=true;

            dfs(i+1,j+1);//j+1表示下一次从上一次往后找，不要往前找，去重
            //走到这里的条件：dfs到达终点或者后面没有符合条件的数
            vis[j]=false;
            num-=a[j];//取消该次所加的数
        }
    }
    return;
}

int main()
{
    cin>>n>>k;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
    }
    dfs(0,0);

    cout<<numm<<endl;

    return 0;
}

优化方法（参数保存当前和）： 具体原因看题解一之前的论述，可以简化代码，不需要写加的过程，直接写到参数，不用减去当前加的值。

第三次优化：

#include <iostream>
#include <cstdio>
#include <string>
#include <string.h>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
typedef long long ll;
#define maxn 10005
#define mod 7654321

int num,n,k,numm;
int a[100];
bool vis[100];

bool isprime(int b)//判断质数
{
    if(b<=1)
    {
        return false;
    }
    for(int i=2;i*i<=b;i++)
    {
        if(b%i==0)
            return false;
    }
    return true;
}

void dfs(int i,int start,int sum)
{
    if(i>=k)
    {
        if(isprime(sum))
            numm++;
        return;
    }
    for(int j=start;j<n;j++)//关键去重，另 j=start
    {
        if(!vis[j])
        {
            vis[j]=true;

            dfs(i+1,j+1,sum+a[j]);//j+1表示下一次从上一次往后找，不要往前找，去重
            //走到这里的条件：dfs到达终点或者后面没有符合条件的数
            vis[j]=false;//到达这里进行下一次for循环，同时将此时的j标记为false,找下一种情况
        }
    }
    return;
}

int main()
{
    cin>>n>>k;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
    }
    dfs(0,0,0);

    cout<<numm<<endl;

    return 0;
}

优化方法（简化代码）： 解释,若此次被用过，直接continue,终止此次直接进行下一次for循环，。

注意：判断质数的函数的函数名最好写成 isprime，最正规的写法，向标准看齐。

if(vis[j])
            continue;

第四次优化：

#include <iostream>
#include <cstdio>
#include <string>
#include <string.h>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
typedef long long ll;
#define maxn 10005
#define mod 7654321

int num,n,k,numm;
int a[100];
bool vis[100];

bool isprime(int b)//判断质数
{
    if(b<=1)
    {
        return false;
    }
    for(int i=2;i*i<=b;i++)
    {
        if(b%i==0)
            return false;
    }
    return true;
}

void dfs(int i,int start,int sum)
{
    if(i>=k)
    {
        if(isprime(sum))
            numm++;
        return;
    }
    for(int j=start;j<n;j++)//关键去重，另 j=start
    {
        if(vis[j])
            continue;
            
        vis[j]=true;
        dfs(i+1,j+1,sum+a[j]);
        vis[j]=false;
    }
    return;
}

int main()
{
    cin>>n>>k;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
    }
    dfs(0,0,0);

    cout<<numm<<endl;

    return 0;
}