Prime Ring Problem UVA - 524(dfs)

题目大意:给你一个n要你输入每个数的两两之间的和是素数有多少种组合方法,第一个数永远是1.
思路:就是求全排列不过加了一个限制条件就是两两之间的和是素数,这个很好解决。
不过我感觉我的代码有点丑了。。。。。输出格式有点坑,最后都是没有空格的。。。。
代码:

#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;

int path[20];
int vis[20];
int n;
bool isprime(int sum){
	for(int i=2;i*i<=sum;i++){
		if(sum%i==0)return false;
	}return true;
}
void dfs(int dep){
	if(dep==n){
		cout<<path[0];
		for(int i=1;i<n;i++)
		cout<<" "<<path[i]; 
		cout<<endl;
		return ;
	}
	for(int i=1;i<=n;i++){
		if(dep==0){
			if(i==1){
				if(!vis[i]){
					vis[i]=1;
					path[dep]=1;
					dfs(dep+1);
				}
			}
		}else if(dep==n-1){
			if(!vis[i]){
				if(isprime(i+path[0])&&isprime(i+path[dep-1])){
					vis[i]=1;
					path[dep]=i;
					dfs(dep+1);
					vis[i]=0;
				}
			}
		}else{
			if(!vis[i]){
				if(isprime(i+path[dep-1])){
					vis[i]=1;
					path[dep]=i;
					dfs(dep+1);
					vis[i]=0;
				}
			}
		}
	}
}
int main(){
	int k=1;int flag=0;
	while(scanf("%d",&n)!=EOF){
		if(flag)cout<<endl;
		flag=1;
		memset(vis,0,sizeof(vis));
		memset(path,0,sizeof(vis));
		cout<<"Case "<<k++<<":"<<endl;
		dfs(0);

	}
}
全部评论

相关推荐

少年郎as:这不把公司名贴出来那我可要喷你了哦
点赞 评论 收藏
分享
浩浩没烦恼:一二面加起来才一个小时? 我一面就一个小时多了
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务