LeetCode 1175. 质数排列

示例 1：

输入：n = 5
输出：12
解释：举个例子，[1,2,5,4,3] 是一个有效的排列，
但 [5,2,3,4,1] 不是，因为在第二种情况里质数 5 被错误地放在索引为 1 的位置上。

示例 2：

输入：n = 100
输出：682289015

提示：

1 <= n <= 100

class Solution {
public:
    int numPrimeArrangements(int n) {
        if(n == 1)
        	return 1;
        int count = 0, i, j;
        unsigned long long ans = 1;
        bool flag;
        for(i = 2; i <= n; ++i)
        {
            flag = true;
            for(j = 2; j <= sqrt(i); ++j)
            {
                if(i%j == 0)
                {
                    flag = false;
                    break;
                }
            }
            if(flag)
                ++count;
        }
        n = n-count;
        while(n)
        {
            ans *= n--;
            ans %= (long long)(1e9 + 7);
        }
        while(count)
        {
            ans *= count--;
            ans %= (long long)(1e9 + 7);
        }
        return ans;
    }
};