选数(dfs,isprime)

题目:P1036 [NOIP2002 普及组] 选数 - 洛谷 | 计算机科学教育新生态 (luogu.com​​​​​​.cn)

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
int n,k;
int a[22];
long long ans;
bool isprime(int n){
	for(int i=2;i<=sqrt(n);i++){
		if(n%i==0) return false;
	}
	return true;
}

void dfs(int m,int sum,int startx){
	if(m==k){ //m表选数 
		if(isprime(sum)) //当前的和 
		ans++; //符合条件加一 
		return ; //跳到下一循环 
	}
	//startx表示升序排列,以免算重 
    for(int i=startx;i<n;i++)
    dfs(m+1,sum+a[i],i+1); //递归 
    return ;
}

signed main(){
	cin>>n>>k;
	for(int i=0;i<n;i++) cin>>a[i];
	dfs(0,0,0);
	cout<<ans<<endl;
	return 0;
}

从这道题学会的:

  • 深度优先遍历(startx以免算重)
  • 在dfs中return; 是指返回主函数,而主函数中有dfs(0,0,0),则此时dfs有开始重新计数;
相关推荐
khalil102014 分钟前
代码随想录算法训练营Day-46 动态规划13 | 647. 回文子串、516.最长回文子序列、动态规划总结
数据结构·c++·算法·leetcode·动态规划·回文子串·回文子序列
学习3人组21 分钟前
柔性排产时序算法+中间过程+阶段目标 细化表格
算法·mes
he___H29 分钟前
算法快与慢--哈希+双指针
算法·leetcode·哈希算法
呃呃本30 分钟前
算法题(回溯)
算法
刀法如飞41 分钟前
Rust数组去重的20种实现方式,AI时代用不同思路解决问题
人工智能·算法·ai编程
yxc_inspire1 小时前
25年CCPC福建邀请赛补题
学习·算法
Raink老师1 小时前
用100道题拿下你的算法面试(链表篇-4):合并 K 个有序链表
算法·链表·面试
Mr_pyx1 小时前
【LeetHOT100】二叉树的中序遍历——Java多解法详解
java·开发语言·深度优先
Liangwei Lin1 小时前
LeetCode 20. 有效的括号
算法
IronMurphy2 小时前
【算法四十四】322. 零钱兑换
算法