1. 问题描述:
请你帮忙给从 1 到 n 的数设计排列方案,使得所有的「质数」都应该被放在「质数索引」(索引从 1 开始)上;你需要返回可能的方案总数。让我们一起来回顾一下「质数」:质数一定是大于 1 的,并且不能用两个小于它的正整数的乘积来表示。由于答案可能会很大,所以请你返回答案模 mod 10^9 + 7 之后的结果即可。
示例 1:
输入:n = 5
输出:12
解释:举个例子,[1,2,5,4,3] 是一个有效的排列,但 [5,2,3,4,1] 不是,因为在第二种情况里质数 5 被错误地放在索引为 1 的位置上。
示例 2:
输入:n = 100
输出:682289015
提示:
1 <= n <= 100
2. 思路分析:
① 因为我们是需要将质数放在质数索引对应的位置,而质数与质数索引是对应起来的,所以第一步我们是需要计算出质数的个数,而这些质数是可以在这些位置进行排列组合的,所以这些质数排列好的种类数是b!(假设质数的个数为b),而剩下来的n - b个不是质数的数字那么又可以排列组合,所以总的种类数应该是两个种类数相乘的结果,注意在计算的时候需要取余防止数字太大溢出
② 这道题目的本质实际上是计算出质数的个数
3. 代码如下:
class Solution:
# 判断是否是素数
def isPrime(self, n):
i = 2
while i * i <= n:
if n % i == 0: return False
i += 1
return True
# 计算n!
def cal(self, n, mod):
res = 1
for i in range(1, n + 1):
res *= i
res %= mod
return res
def numPrimeArrangements(self, n: int) -> int:
# 非质数的数目阶乘与质数的数目阶乘相乘的结果就是答案的数目, 所以这道题目的关键是计算出质数的个数
count, mod = 0, 10 ** 9 + 7
for i in range(2, n + 1):
if self.isPrime(i): count += 1
return self.cal(count, mod) % mod * self.cal(n - count, mod) % mod