Java 洛谷P1036 选数

最新推荐文章于 2022-11-29 20:44:04 发布

小崔崔谁用的

最新推荐文章于 2022-11-29 20:44:04 发布

阅读量142

点赞数

分类专栏：洛谷 java 文章标签： dfs 算法剪枝 java

本文链接：https://blog.csdn.net/weixin_45665566/article/details/113785602

版权

java 同时被 2 个专栏收录

75 篇文章 1 订阅

订阅专栏

洛谷

63 篇文章 1 订阅

订阅专栏

这篇博客介绍了一个用Java编写的程序，该程序从输入数组中通过深度优先搜索找到长度为k的子序列，这些子序列的和是质数。程序包含了`isPrime`函数来判断一个数是否为质数，并通过递归的`dfs`方法遍历所有可能的子序列。博主展示了如何在给定约束下进行数值计算和优化算法设计。

摘要由CSDN通过智能技术生成

题目地址
解析可以看另一篇题解
PERKET
下面上AC代码

import java.util.*;
import java.math.*;
public class Main {
    public static List<Integer> list=new ArrayList<Integer>();
    public static int res=0;
    public static void main(String[] args){
    	Scanner sc=new Scanner(System.in);
    	int n=sc.nextInt();
    	int k=sc.nextInt();
    	int[] nums=new int[n];
    	for(int i=0;i<n;i++)
    		nums[i]=sc.nextInt();
    	dfs(nums,0,k);
    	System.out.println(res);
    }
    public static void dfs(int[] nums,int n,int k){
    	int len=nums.length;
    	if(list.size()==k){
    		int cur=0;
    		for(Integer i:list)
    			cur+=i;
    		if(isPrime(cur)==1)
    			res++;
    		return;
    	}
    	if(list.size()+len-n+1<k||n==len)
    		return;
    	list.add(nums[n]);
    	dfs(nums,n+1,k);
    	list.remove(new Integer(nums[n]));
    	dfs(nums,n+1,k);
    	
    }
    public static int isPrime(int n){
    	if(n==1)
    		return 0;
    	if(n==2||n==3)
    		return 1;
    	if(n%6!=1&&n%6!=5)
    		return 0;
    	for(int i=5;i<=Math.sqrt(n);i+=6){
    		if(n%i==0||n%(i+2)==0)
    			return 0;
    	}
    	return 1;
    }
}