选数(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有开始重新计数;
相关推荐
带多刺的玫瑰1 小时前
Leecode#9刷题之回文数
数据结构·算法
泯泷1 小时前
手搓JSVM第 5 篇:写第一个编译器:从 AST 生成 IR
前端·javascript·算法
泯泷1 小时前
手搓JSVM第 7 篇:控制流:if、while 与 jump
前端·javascript·算法
泯泷1 小时前
手搓JSVM第 6 篇:把 IR 编成字节码:emit 与 label fixup
前端·javascript·算法
泯泷1 小时前
手搓JSVM第 3 篇:从栈式 VM 到寄存器式 VM:为什么我们选择寄存器
前端·javascript·算法
泯泷1 小时前
第 4 篇:让 VM 支持变量:Slot、Environment 与 TDZ
前端·javascript·算法
Asize2 小时前
54. 螺旋矩阵
算法
Asize2 小时前
73. 矩阵置零
算法
微露清风3 小时前
快慢指针算法学习记录
学习·算法·快慢指针
benchmark_cc3 小时前
1000只ETF的5分钟K线如何批量获取?QuantDash分页策略与高性能Python实践
开发语言·人工智能·爬虫·python·算法·quantdash·量化数据源