刷官方题单,刷完几个黄题飘了,觉得这种橙体不在话下,于是乎就卡住了。
觉得自己对于dfs较为基础的部分理解还在不够深刻,以至于在一些十分基础的地方犯错,因此写此博客来作为总结。
题目描述
代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 25;
int n, k, ans, a[N];
bool is_prime(int x)
{
if(x < 2)
return false;
for(int i = 2; i <= x / i; i ++ )
if(x % i == 0)
return false;
return true;
}
void dfs(int path, int sum, int st)
{
if(path > k)
{
if(is_prime(sum))
ans ++ ;
return;
}
for(int i = st; i <= n; i ++ )
dfs(path + 1, sum + a[i], i + 1);
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(NULL); cout.tie(NULL);
cin >> n >> k;
for(int i = 1; i <= n; i ++ )
cin >> a[i];
dfs(1, 0, 1);
cout << ans << endl;
return 0;
}
对于该代码需要总结的地方
1.对于dfs基础模板理解的不够深刻,前期用vis数组标记来做,实质上每次做dfs循环的时候都会从头开始遍历,故而就达不到了所先想要的效果。
2.对于dfs传入内容的更新,实际上只要随着我的需要变化的东西都可以当作dfs变量传进去,这样既会节省维护新的数组所需要的时间和空间。
3.为了去重,我需要保证在判断dfs结束的语句之中所涵盖的数组部分必然是不一致的。所以只需保证每次都增加下标直至满足条件为止,这样就能保证cnx。