前言
省赛之后,好长一段时间没有做过题了,本来想找道水题热热身,没想到这么一道题目都被折磨了挺久orz,真的是不能松懈啊,本来预定的省赛赛后感也鸽了…
题目
原题链接:https://www.luogu.org/problemnew/show/P1036
P1036: 选数
Time Limit: 1 Sec
Memory Limit: 128 Mb
题目描述
已知 nnn 个整数 x1,x2,…,xnx_1,x_2,…,x_nx1,x2,…,xn,以及111个整数kkk(k<nk<nk<n)。从nnn个整数中任选kkk个整数相加,可分别得到一系列的和。例如当n=4,k=3n=4,k=3n=4,k=3,444个整数分别为3,7,12,193,7,12,193,7,12,19时,可得全部的组合与它们的和为:
3+7+12=223+7+12=223+7+12=22
3+7+19=293+7+19=293+7+19=29
7+12+19=387+12+19=387+12+19=38
3+12+19=343+12+19=343+12+19=34。
现在,要求你计算出和为素数共有多少种。
例如上例,只有一种的和为素数:3+7+19=293+7+19=293+7+19=29。
输入输出格式
输入格式:
键盘输入,格式为:
n,kn,kn,k(1≤n≤20,k<n1 \le n \le 20,k<n1≤n≤20,k<n)
x1,x2,…,xn(1≤xi≤5000000)x_1,x_2,…,x_n (1 \le x_i \le 5000000)x1,x2,…,xn(1≤xi≤5000000)
输出格式:
屏幕输出,格式为: 111个整数(满足条件的种数)。
输入输出样例
输入样例#1:
4 3
3 7 12 19
输出样例#1:
1
解题思路
其实这是一道挺简单的水题(可能太久没做题脑回路生锈了吧orz),思路挺直接的——按照题目要求来就行,求一组数中特定个数的数的素数和数目,暴力直接点的方法的话——for循环大法·十式嵌套!
当然,用递归来解,写起来会简单省事的多,那么直接来看代码吧。
解题代码
import java.math.BigInteger;
import java.util.Scanner;
public class Main {
static int n,k,ans = 0,a[],temp = 0;
static boolean f[] = new boolean[21];
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
while(sc.hasNext()){
n = sc.nextInt();
k = sc.nextInt();
a = new int [n+1];
for(int i = 1;i <= n;i++){
a[i] = sc.nextInt();
f[i] = true;
}
check(1,1);
System.out.println(ans);
ans = 0;
temp = 0;
}
}
public static void check(int x,int y){
for(int j=y;j<=n;j++){
if(f[j]){
temp+=a[j];
f[j]=false;
if(x==k&&BigInteger.valueOf(temp).isProbablePrime(23))
ans++;
else check(x+1,j+1);
temp-=a[j];
f[j]=true;
}
}
}
}
終わりました、これからも頑張りましょう!