前言
这几天做了不少普及题感觉自己还是有进步的(相对大一时候的我,那时候不敢做普及题哈哈哈)
依旧是先放题目来源:P1036 [NOIP2002 普及组] 选数
解题思路:
这题如果是个数学题的话,其实就是先找出n个数中k个数组成的不同组合,然后计算加和,判断加和是不是素数就行了。但是在这里需要编程解决,因此,主要解决的就是两个问题,一个是搜索所有k个数的组合;另一个是判断这k个数的加和是否是素数。判断是不是素数是个基本功我就不说了。然后就是搜索k个数的组合。在这里,我采用的是递归的思想。具体来说就是,从第一个数开始,会面临两个选择,一个选择是取第一个数,然后在剩下的第2到第n个数中找k-1个数;另一个是不选这个数,然后在第2到第n个数中找k个数;对于第二个数来说,承接上面的讨论,它也面临同样的选择。这就很容易看出来是个递归的思路了。即在第left到第right个a[]数组中找第k个数,目前加和为sum;当k为0时,表示已经找完了,所以直接判断当前加和是否是素数,是就返回1不是就返回0;当k>0并且left>=right时,表明此时没有选完最初要的k个数,且left>=right保证了当前还可以选,因此在数组a[]的left+1到right项中选第k-1个数,当前加和为sum+a[left],加上不选当前的数,也就是在数组a[]中的left+1到right项中选第k个数,当前加和还是sum然后返回两种情况的数量和num=num_left+num_right;若left>right说明数组已经被选完了又没有满足前面的条件,因此直接返回0就可以。
下面给出我的AC代码:
#include <stdio.h>
#include <stdlib.h>
int is_prime(int a){
int i;
if(a==1)
return 0;
for(i=2;i*i<=a;i++){
if(a%i==0)
return 0;
}
return 1;
}
int search(int left,int right,int k,int a[],int sum){
int num,num1,num2;
if(k==0){
if(is_prime(sum))
return 1;
else
return 0;
}
else if(k>0&&left<=right){
num1=search(left+1,right,k-1,a,sum+a[left]);
num2=search(left+1,right,k,a,sum);
num=num1+num2;
}else if(left>right){
return 0;
}
return num;
}
int main(int argc, char *argv[]) {
int i,n,k;
scanf("%d%d",&n,&k);
int a[n];
for(i=0;i<n;i++)
scanf("%d",&a[i]);
int num,sum=0;
num=search(0,n-1,k,a,sum);
printf("%d",num);
return 0;
}