入门力扣自学笔记78 C++ （题目编号1175）

最新推荐文章于 2023-12-22 08:01:33 发布

DK_Sorhic

最新推荐文章于 2023-12-22 08:01:33 发布

阅读量150

点赞数

分类专栏：力扣算法学习文章标签： leetcode 算法 c++

本文链接：https://blog.csdn.net/DK_Sorhic/article/details/125536713

版权

力扣算法学习专栏收录该内容

280 篇文章 7 订阅

订阅专栏

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/problems/prime-arrangements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

这道题只需要先判断出来质数，完后假设n以内的质数为a个，非质数为b个，那么进行一次排列组合，总个数为 $A_{a}^{a}$ + $A_{b}^{b}$ ，返回最后结果即可。

注意，1不是质数。

代码：

class Solution {
public:
    int numPrimeArrangements(int n) {
        long long ans = 1;
        vector<bool> hasv(n+1, false);
        vector<int> primes;
        for(int i = 2; i <= n; i++){
            if(hasv[i]) continue; 
            primes.push_back(i);
            for(int k = 1; ; k++){
                int cur = i * k;
                if(cur <= n) hasv[cur] = true;
                else break;
            }
        }
        int mod = 1e9+7;
        for(int i = 2; i <= primes.size(); i++) 
            ans = ans * i % mod;
        for(int i = 2; i <= n - primes.size(); i++) 
            ans = ans * i % mod; 
        return ans;
    }
};