HDU - 1016 dfs

最新推荐文章于 2021-02-26 15:30:16 发布

原创最新推荐文章于 2021-02-26 15:30:16 发布 · 358 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#搜索

算法之路专栏收录该内容

240 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）算法解决质数环问题的方法，并提供了完整的AC代码示例。针对输入的整数n，寻找一种排列方式使得任意相邻两数之和均为质数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：经典dfs问题，注意1和最后一个数的和也必须是质数。对于n是奇数的情况，根本不会有结果，因为必定会有两个奇数相邻，他们的和是大于2的偶数就不可能是质数。

AC代码

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <utility>
#include <string>
#include <iostream>
#include <map>
#include <set>
#include <vector>
#include <queue>
#include <stack>
using namespace std;
#pragma comment(linker, "/STACK:1024000000,1024000000") 
#define eps 1e-10
#define inf 0x3f3f3f3f
#define PI pair<int, int> 
typedef long long LL;
const int maxn = 20 + 5;
int vis[maxn], ans[maxn], d[maxn], n;
void init(int n) {
	int m = sqrt(n+0.5);
	memset(vis, 0, sizeof(vis));
	for(int i = 2; i <= m; ++i) if(!vis[i])
		for(int j = i*i; j <= n; j+=i) vis[j] = 1;
}

void dfs(int cnt) {
	if(cnt == n) {
		if(!vis[ans[n-1]+1]) for(int i = 0; i < n; ++i) {
			if(i==n-1) printf("%d\n", ans[i]);
			else printf("%d ", ans[i]);
		}
		return;
	}
	for(int i = 2; i <= n; ++i) {
		if(!d[i] && !vis[ans[cnt-1]+i]) {
			d[i] = 1;
			ans[cnt] = i;
			dfs(cnt+1);
			d[i] = 0;
		}
	}
}

int main() {
	init(22);
	int kase = 0;
	while(scanf("%d", &n) == 1) {
		memset(d, 0, sizeof(d));
		printf("Case %d:\n", ++kase);
		ans[0] = 1;
		d[1] = 1;
		dfs(1);
		printf("\n");
	}
	return 0;
}

如有不当之处欢迎指出！