Leecode(30)1175.质数排列-简单

请你帮忙给从 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

统计n以内素数的个数，答案由素数的全排列数乘上非素数的全排列数得到。

因此我们只需要统计素数的个数即可

class Solution:
    def numPrimeArrangements(self, n: int) -> int:
        MOD = 10**9 + 7
        
        # Function to determine if a number is prime
        def is_prime(num):
            if num < 2:
                return False
            if num == 2:
                return True
            if num % 2 == 0:
                return False
            for i in range(3, int(num**0.5) + 1, 2):
                if num % i == 0:
                    return False
            return True
        
        prime_count = sum(1 for i in range(1, n + 1) if is_prime(i))
        non_prime_count = n - prime_count
        
        # Calculate factorial of prime_count and non_prime_count modulo MOD（全排列）
        def factorial(num):
            if num <= 1:
                return 1
            result = 1
            for i in range(2, num + 1):
                result = (result * i) % MOD
            return result
        
        prime_arrangements = factorial(prime_count)
        non_prime_arrangements = factorial(non_prime_count)
        
        return (prime_arrangements * non_prime_arrangements) % MOD

1. 判断质数的函数：is_prime 函数用于判断一个数是否为质数。

2. 计算质数和非质数的个数：使用了 is_prime 函数来统计 1 到 n 中质数和非质数的个数。

3. 阶乘计算函数：factorial 函数用于计算阶乘，并在计算过程中取模防止溢出。

4. 计算结果：计算质数和非质数的排列数，然后将它们相乘并取模得到最终结果。