洛谷 P1036 递归 选数

洛谷 P1036 递归 选数

#include <stdio.h>
#include <stdlib.h>

int n,k,count=0;
int a[22]={0};

void Fun(int number,int sum,int start);/*递归函数*/
int Judge(int n);/*判断质数*/

int main()
{
    int i;
    scanf("%d %d",&n,&k);/*输入*/
    for(i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    Fun(0,0,1);        /*递归调用*/
    printf("%d",count);/*输出*/

    return 0;
}

void Fun(int number,int sum,int start)/*number是已加的数的个数，
                            sum是加和，start是选数开始位置*/
{
    int i;
    int flag[30]={0};/*标记一个数是否被用过，用过为1*/
    if(number==k)/*若已选了n个数*/
    {
        if(Judge(sum)){  /*且加和为质数*/
            count++;/*计数+1*/
        }
    }
    for(i=start;i<=n;i++)
    {
        if(flag[i]){   /*用过该数就跳过*/
            continue;
        }
        flag[i]=1;    /*把该数标记*/
        Fun(number+1,sum+a[i],i+1);/*若没到k则选出下一个数*/
        flag[i]=0;    /*这个数OK了，+1换下一个数*/
    }
}
int Judge(int n)
{
    int i;
    for(i=2;i*i<=n;i++)
    {
        if(n%i==0)
        {
            return 0;
        }

    }
    return 1;
}