NOIP 2002 普及组 复赛 选数

NOIP 2002 普及组 复赛 选数

//洛谷 p1036 选数

//难度：普及-

//考点：输入，输出 ,递归，排列组合，质数的判定，深度优先遍历,阶乘计算  

//适用：初中

//小技巧：深度优先遍历，数据有重复，别忘了/f(k) 阶乘

#include <stdio.h>

#include <string.h>

int a[30],b[30],vis[30];

int n,k;

int count=0;

int f(int n){//阶乘

    if(n==1)

        return 1;

    return f(n-1)*n;

}

int isprime(int n){//0非质数,1质数

    int i;

    if(n==1)

        return 0;

    if(n==2)

        return 1;

    for(i=2;i*i<=n;i++)

        if(n%i==0)

            return 0;

    return 1;

}

void dfs(int step){

    int i,j,sum;

    if(step==k+1){

        sum=0;

        for(j=1;j<=k;j++)

            sum+=a[j];

        if(isprime(sum))

            count++;    

        return ;

    }

    for(i=1;i<=n;i++)

        if(vis[i]==0){

            vis[i]=1;

            a[step]=b[i];

            dfs(step+1);

            vis[i]=0;

        }

}

int main(){

    int i;

    memset(vis,0,sizeof(vis));

    scanf("%d%d",&n,&k);

    for(i=1;i<=n;i++)

        scanf("%d",&b[i]);

    dfs(1);

    printf("%d\n",count/f(k));

    return 0;

}