选数(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有开始重新计数;
相关推荐
寻星探路5 分钟前
【算法专题】滑动窗口:从“无重复字符”到“字母异位词”的深度剖析
java·开发语言·c++·人工智能·python·算法·ai
wen__xvn44 分钟前
代码随想录算法训练营DAY14第六章 二叉树 part02
数据结构·算法·leetcode
Ka1Yan1 小时前
[数组] - 代码随想录(2-6)
数据结构·算法·leetcode
漫随流水2 小时前
leetcode算法(104.二叉树的最大深度)
数据结构·算法·leetcode·二叉树
机器学习之心HML2 小时前
鲸鱼算法(WOA)优化Kriging模型
算法
DYS_房东的猫2 小时前
《 C++ 零基础入门教程》第6章:模板与 STL 算法 —— 写一次,用万次
开发语言·c++·算法
Tim_102 小时前
【算法专题训练】37、前缀树&二叉树
算法
NineData2 小时前
第三届数据库编程大赛-八强决赛成绩揭晓
数据库·算法·代码规范
雍凉明月夜2 小时前
深度学习之目标检测yolo算法Ⅱ(v4)
深度学习·算法·yolo·目标检测
FAFU_kyp3 小时前
Rust 的 引用与借用
开发语言·算法·rust