洛谷P1036

题目

题目链接：https://www.luogu.com.cn/problem/P1036

没什么好说的 我好菜 不会递归

代码1 dfs

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define ll long long
using namespace std;
const int INF = 0x3f3f3f3f;
int t,k;
bool vis[100];
int a[100];
bool prime(int n){
	bool flag=1;
	for(int i=2;i<=sqrt(n);i++){
		if(n%i==0) flag=0;
	}
	return flag;
}
int res=0;
void dfs(int pos,int sum,int k){
	if(pos>t) {
		if(k==0) res+=prime(sum);
		return;
	}
	dfs(pos+1,sum+a[pos],k-1);
	dfs(pos+1,sum,k);
} 
int main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	cin>>t>>k;
	for(int i=1;i<=t;i++){
		cin>>a[i];
	}
	dfs(1,0,k);
	cout<<res<<endl;
    return 0;
}

代码2 优化

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define ll long long
using namespace std;
const int INF = 0x3f3f3f3f;
int a[100];	int t,k;
bool prime(int n){
	bool flag=1;
	for(int i=2;i<=sqrt(n);i++){
		if(n%i==0) flag=0;
	}
	return flag;
}
int f(int k,int sum,int l,int r){
	if(k==0) return prime(sum);
	int res=0;
	for(int i=l;i<=r;i++){
		res+=f(k-1,sum+a[i],i+1,r);
	}
	return res; 
}
int main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	cin>>t>>k;
	for(int i=1;i<=t;i++){
		cin>>a[i];
	}
	cout<<f(k,0,1,t)<<endl;
    return 0;
}