前言:
在学校信息队训练的时候布置了这一道题目,但是只求一次。然后上洛谷发现UVA题库求多次,因为方法一样只需嵌套,所以就做了一下(没有账号提交不了=m=)
信息队原题
原题图片
1.题目分析
题目要求将整数 $1$ 到 $n$ 组成一个环,使得相邻的两个整数之和均为素数。
我们需要找到所有满足条件的素数环,并按照要求输出。
由于 $n$ 的值不超过 $16$ ,我们可以使用回溯法来尝试所有可能的组合。
2.做题思路
判断当前取出的数字是否和前一个组成素数,这一步可以用一个 check
函数来解决(num的值为两个数字的和)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
| bool check(int num) { if (num < 2) { return false; } for (int i = 2; i * i <= num; i++) { if (num % i == 0) { return false; } } return true; }
|
在创建一个 back
函数用于回溯:
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
| void back(int cur) { if (cur == n) { if (check(arr[0] + arr[n - 1])) { for (int i = 0; i < n; i++) { cout << arr[i] << " "; } cout << endl; } return; } for (int i = 2; i <= n; i++) { if (!vis[i] && check(arr[cur - 1] + i)) { vis[i] = true; arr[cur] = i; back(cur + 1); vis[i] = false; } } }
|
在back函数中,首先判断是否已经放置了 $n$ 个数字。如果是,则检查最后一个数字和第一个数字之和是否为素数。如果是素数,则输出这个环。
然后,从数字 $2$ 到 $n$ 依次尝试将每个数字放在环的下一个位置。如果该数字没有被使用过,并且与前一个数字之和为素数,就将其放在环的下一个位置,并标记为已使用,然后继续递归放置下一个数字。
如果放置过程中出现相邻两个数字之和不是素数的情况,就回溯到上一个数字,重新选择下一个数字。
最后是 main
函数,在main函数中,读取输入的 $n$ ,初始化标记数组和环的第一个数字,然后调用 back
函数从数字 $1$ 开始回溯。
3.复杂度计算
时间复杂度:由于需要尝试所有可能的组合,时间复杂度为 $O(n!)$ ,其中 n 是输入的数字个数
空间复杂度:主要是使用了标记数组和环数组,空间复杂度为 $O(n)$
4.完整代码
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
| #include <bits/stdc++.h> using namespace std;
const int MAXN = 16; int n; int arr[MAXN]; bool vis[MAXN];
bool check(int num) { if (num < 2) { return false; } for (int i = 2; i * i <= num; i++) { if (num % i == 0) { return false; } } return true; }
void back(int cur) { if (cur == n) { if (check(arr[0] + arr[n - 1])) { for (int i = 0; i < n; i++) { cout << arr[i] << " "; } cout << endl; } return; } for (int i = 2; i <= n; i++) { if (!vis[i] && check(arr[cur - 1] + i)) { vis[i] = true; arr[cur] = i; back(cur + 1); vis[i] = false; } } }
int main() { cin >> n; memset(vis, false, sizeof(vis)); vis[1] = true; arr[0] = 1; back(1); return 0; }
|
洛谷UVA题库版
前面不是说了吗,UVA也有一个版本,是给出多个值,这里给出题目:
洛谷UVA题目
样例组
INPUT #1
OUTPUT #1
1 2 3 4 5 6 7 8 9
| Case 1: 1 4 3 2 5 6 1 6 5 2 3 4
Case 2: 1 2 3 8 5 6 7 4 1 2 5 8 3 4 7 6 1 4 7 6 5 8 3 2 1 6 7 4 3 8 5 2
|
其实就是多次输入,用 while
循环解决,终止条件为 n == EOF
,所以可以这样写。
1 2 3 4 5 6 7 8
| while(n != EOF) { cin >> n; memset(vis, false, sizeof(vis)); vis[1] = true; arr[0] = 1; back(1); }
|
现在是把刚才信息队版本的输入和操作套在了循环里,这就完了,是不是很简单?
其实没完,题目中说要在第 i
行添加上 Case i
,所以再搞一个变量就完事了!
1 2 3 4 5 6 7 8 9 10
| int idx = 1; while(n != EOF) { cin >> n; cout << "Case " << idx++ << ":" << endl; memset(vis, false, sizeof(vis)); vis[1] = true; arr[0] = 1; back(1); }
|
最终代码
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 65 66 67
| #include <bits/stdc++.h> using namespace std;
const int MAXN = 16; int n; int arr[MAXN]; bool vis[MAXN];
bool check(int num) { if (num < 2) { return false; } for (int i = 2; i * i <= num; i++) { if (num % i == 0) { return false; } } return true; }
void back(int cur) { if (cur == n) { if (check(arr[0] + arr[n - 1])) { for (int i = 0; i < n; i++) { cout << arr[i] << " "; } cout << endl; } return; } for (int i = 2; i <= n; i++) { if (!vis[i] && check(arr[cur - 1] + i)) { vis[i] = true; arr[cur] = i; back(cur + 1); vis[i] = false; } } }
int main() { int idx = 1; while(n != EOF) { cin >> n; cout << "Case " << idx++ << ":" << endl; memset(vis, false, sizeof(vis)); vis[1] = true; arr[0] = 1; back(1); } return 0; }
|
写在最后
到这里本片题解就结束了,UVA版本没提交,前面说了注册不起账号,但是代码应该没问题,有问题评论告知,谢谢支持🙏