萌新初学dfs,感觉懂了原理,但是实在不懂得怎么实现出来,一直在逻辑混乱中
求大佬给个意见!(想用回溯的思路)
#include<bits/stdc++.h>
#include<iostream>
using namespace std;
long long a[100];
int n,k;
int ans=0;
int prime(int t)
{
for(int i=2;i*i<=t;i++)
if(t%i==0)
return 0;
return 1;
}
void dfs(int m,int sum)
{
if(m==k+1)
{
ans+=prime(sum);
return;
}
else
{
for(int i=m;i<=n;i++)
{
sum+=a[i];
dfs(m+1,sum);
sum-=a[i];
}
}
return;
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
dfs(1,0);
printf("%d",ans);
return 0;
}