【题解】洛谷P1036 选数

前言

省赛之后,好长一段时间没有做过题了,本来想找道水题热热身,没想到这么一道题目都被折磨了挺久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;
    		} 
    	}
    }
    
}

終わりました、これからも頑張りましょう!

猜你喜欢

转载自blog.csdn.net/Jourofant/article/details/107496726
今日推荐