【Python】【难度：简单】Leetcode 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
 

提示：

1 <= n <= 100

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/prime-arrangements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

class Solution(object):
    def numPrimeArrangements(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n==1:
            return 1
        
        def helper(n):
            if n==1:
                return 1
            else:
                return n*helper(n-1)


        count=1
        for i in range(3,n+1,2):
            k=3
            flag=True
            while k<i:
                if i%k==0:
                    flag=False
                k+=2
            if flag:
                count+=1
        return (helper(n-count)*helper(count))%(10**9+7)

 

执行结果：

通过

显示详情

执行用时 :20 ms, 在所有 Python 提交中击败了90.91%的用户

内存消耗 :12.7 MB, 在所有 Python 提交中击败了50.00%的用户