ZOJ-2203-Anti-prime Sequences(搜索)

题目链接

ZOJ-2203

题意

求一个 1-n 的排列,使得任意相邻 2-d 个数之和均为素数

思路

直接按照题意暴搜即可,注意提前打好素数表,提高判断素数的效率
注意回溯,搜到答案及时return

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#include <bits/stdc++.h>
#include <iostream>
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
const int maxn = 10000+10;
const int mod = 1e9+7;
const double eps = 1e-10;
int n , m , d;
int prime[maxn];
int vis[maxn];
void init()
{
prime[0] = prime[1] = 0;
for(int i = 2 ; i < maxn ; i ++)prime[i] = 1;
for(int i = 2 ; i < maxn ; i ++){
if(prime[i]){
for(int j = i + i ; j < maxn ; j += i)
prime[j] = 0;
}
}
}
int a[maxn] ;
vector<int> ans;
bool check(int pos , int x)
{
int sum = x;
for(int i = pos-1 ; i >= 0 && i > pos - d ; i --){
sum += ans[i];
if(prime[sum])return false;
}
return true;
}
bool ok;
bool dfs(int pos)
{
if(ans.size() == m-n + 1)return true;
for(int i = n ; i <= m ; i ++){
if(!vis[i] && check(pos , i)){
vis[i] = 1;
ans.push_back(i);
if(dfs(pos + 1))return true;
vis[i] = 0;
ans.pop_back();
}
}
return false;
}
int main(int argc, char const *argv[])
{
init();
while(cin >> n >> m >> d && (n+m+d)){
ans.clear();
memset(vis , 0 , sizeof(vis));
ok = false;
if(dfs(0)){
for(int i = 0 ; i <= m-n ; i ++){
printf("%d%c" , ans[i] , i == m-n ? '\n' : ',');
}
}else cout<<"No anti-prime sequence exists."<<endl;
}
return 0;
}