「题解」 素数环

磅豆龙 融会贯通

前言:

在学校信息队训练的时候布置了这一道题目,但是只求一次。然后上洛谷发现UVA题库求多次,因为方法一样只需嵌套,所以就做了一下(没有账号提交不了=m=)

信息队原题

原题图片

1.题目分析

题目要求将整数组成一个环,使得相邻的两个整数之和均为素数。

我们需要找到所有满足条件的素数环,并按照要求输出。

由于的值不超过,我们可以使用回溯法来尝试所有可能的组合。

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)  // 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函数中,首先判断是否已经放置了个数字。如果是,则检查最后一个数字和第一个数字之和是否为素数。如果是素数,则输出这个环。

然后,从数字依次尝试将每个数字放在环的下一个位置。如果该数字没有被使用过,并且与前一个数字之和为素数,就将其放在环的下一个位置,并标记为已使用,然后继续递归放置下一个数字。

如果放置过程中出现相邻两个数字之和不是素数的情况,就回溯到上一个数字,重新选择下一个数字。

最后是 main 函数,在main函数中,读取输入的,初始化标记数组和环的第一个数字,然后调用 back 函数从数字开始回溯。

3.复杂度计算

时间复杂度:由于需要尝试所有可能的组合,时间复杂度为,其中 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;
}

// 回溯
// cur表示当前位置
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

1
1
2
6
8

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;  // 此处 idx 变量即 i 变量
while(n != EOF)
{
cin >> n;
cout << "Case " << idx++ << ":" << endl; // 输出 Case i 因为 "++" 在后是先赋值再运算,所以可以这么写,就相当于先输出 idx 当前值再加
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;
}

// 回溯
// cur表示当前位置
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版本没提交,前面说了注册不起账号,但是代码应该没问题,有问题评论告知,谢谢支持🙏

  • Title: 「题解」 素数环
  • Author: 磅豆龙
  • Created at : 2024-06-27 13:25:16
  • Updated at : 2024-08-03 12:45:10
  • Link: https://blog.setbun.com/p/20240627.html
  • License: This work is licensed under CC BY-NC-SA 4.0.
Comments