小Q的非素数和排列问题 ｜ 豆包MarsCode AI 刷题

题目解析

题目描述

小Q对排列很感兴趣，她想知道有多少个长度为 n 的排列满足任意两个相邻元素之和都不是素数。排列定义为一个长度为 n 的数组，其中包含从1到 n 的所有整数，每个数字恰好出现一次。

思路

1. 生成所有排列：我们需要生成所有可能的排列。可以使用 itertools.permutations 来生成所有排列。
2. 检查相邻元素之和是否为素数：对于每个排列，检查相邻元素之和是否为素数。可以使用一个辅助函数来判断一个数是否为素数。
3. 统计符合条件的排列：统计所有符合条件的排列的数量。

图解

假设 n = 4，我们需要生成所有长度为4的排列，并检查每个排列的相邻元素之和是否为素数。

1. 生成排列：

   • 例如，排列 [1, 2, 3, 4]
   • 所有排列：[1, 2, 3, 4], [1, 2, 4, 3], [1, 3, 2, 4], ...

2. 检查相邻元素之和：

   • 对于排列 [1, 2, 3, 4]，检查 1+2, 2+3, 3+4 是否为素数。
   • 如果所有相邻元素之和都不是素数，则该排列符合条件。

3. 统计符合条件的排列：

   • 统计所有符合条件的排列的数量。

代码详解

def is_prime(num):
    # 判断一个数是否为素数的辅助函数
    if num <= 1:
        return False
    for i in range(2, int(num**0.5) + 1):
        if num % i == 0:
            return False
    return True

def solution(n: int) -> int:
    # 生成所有排列
    all_permutations = permutations(range(1, n + 1))
    
    # 统计符合条件的排列数量
    count = 0
    for perm in all_permutations:
        valid = True
        for i in range(n - 1):
            if is_prime(perm[i] + perm[i + 1]):
                valid = False
                break
        if valid:
            count += 1
    
    return count

if __name__ == '__main__':
    print(solution(n = 5) == 4)
    print(solution(n = 3) == 0)
    print(solution(n = 6) == 24)

代码解释

1. is_prime 函数：

   • 用于判断一个数是否为素数。
   • 如果数小于等于1，直接返回 False。
   • 否则，检查从2到 sqrt(num) 之间的数是否能整除 num，如果能，则返回 False，否则返回 True。

2. solution 函数：

   • 生成所有长度为 n 的排列。
   • 遍历每个排列，检查相邻元素之和是否为素数。
   • 如果所有相邻元素之和都不是素数，则计数器加一。

3. 主程序：

   • 调用 solution 函数并打印结果，验证是否符合预期。

总结与感悟

• 排列生成：使用 itertools.permutations 可以方便地生成所有排列。
• 素数判断：素数判断是一个常见的算法问题，可以通过简单的循环来实现。
• 排列检查：对于每个排列，需要检查相邻元素之和是否为素数，这需要遍历排列并调用素数判断函数。
• 时间复杂度：生成所有排列的时间复杂度为 O(n!)，检查每个排列的时间复杂度为 O(n)，因此总的时间复杂度为 O(n * n!)。

通过这道题目，我们可以更好地理解排列的生成和素数的判断，同时也需要注意算法的时间复杂度，特别是在 n 较大时，可能需要优化算法或使用更高效的算法来解决问题。