题目描述:
请你帮忙给从 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
方法1:
主要思路:
(1)统计出 n 范围内的所有的素数,然后将素数和非素数的阶乘的成绩对1000000007求余即可;
class Solution {
public:
//判断是否为素数,直接跳过了偶数的判断
bool is_prime(int n){
for(int i=3;i*i<=n;i+=2){
if(n%i==0){
return false;
}
}
return true;
}
//计算阶乘
long fun(int n){
long res=1;
while(n>1){
res*=n;
res%=1000000007;
--n;
}
return res;
}
int numPrimeArrangements(int n) {
if(n<3){
return 1;
}
int nums_prime=1;
//统计素数,跳过偶数直接判断
for(int i=3;i<=n;i+=2){
if(is_prime(i)){
++nums_prime;
}
}
//计算
return fun(nums_prime)*fun(n-nums_prime)%(1000000007);
}
};