求助QAQ
查看原帖
求助QAQ
160870
迟陌楼主2023/7/21 23:24

萌新初学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;
}
2023/7/21 23:24
加载中...