共计 1908 个字符,预计需要花费 5 分钟才能阅读完成。
题目形容
这是 LeetCode 上的 1175. 质数排列 ,难度为 简略。
Tag :「数学」、「组合数」、「二分」、「打表」
请你帮忙给从 $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$
打表 + 二分 + 数学
依据题意,可将问题转换为求 $n$ 以内的质数个数,记为 $a$,同时可得非质数个数为 $b = n – a$。
质数的搁置计划数为 $a!$,而非质数的搁置计划数为 $b!$,依据「乘法原理」总的搁置计划数为 $a! \times b!$。
咱们能够通过「打表」的形式将 $100$ 以内的质数预处理到数组 list
中,对于每个 $n$ 而言,咱们找到第一个满足「值小于等于 $n$」的地位,从而得悉 $n$ 范畴以内的质数个数。
代码:
class Solution {static int MOD = (int)1e9+7;
static List<Integer> list = new ArrayList<>();
static {for (int i = 2; i <= 100; i++) {
boolean ok = true;
for (int j = 2; j * j <= i; j++) {if (i % j == 0) ok = false;
}
if (ok) list.add(i);
}
}
public int numPrimeArrangements(int n) {int l = 0, r = list.size() - 1;
while (l < r) {
int mid = l + r + 1 >> 1;
if (list.get(mid) <= n) l = mid;
else r = mid - 1;
}
int a = r + 1, b = n - a;
long ans = 1;
for (int i = b; i > 1; i--) ans = ans * i % MOD ;
for (int i = a; i > 1; i--) ans = ans * i % MOD ;
return (int)ans;
}
}
- 工夫复杂度:二分的复杂度为 $O(\log{n})$,计算计划数的复杂度为 $O(n)$。整体复杂度为 $O(n)$
- 空间复杂度:$O(C)$,其中 $C = 25$ 为 $100$ 以内的质数个数
打表 + 数学
更进一步,对于特定的 $n$ 而言,咱们在预处理 $100$ 以内的质数时,曾经能够确定在 $[1, n]$ 内有多少个质数,从而省掉二分操作。
应用数组 cnts
记录下不超过以后值范畴内质数的个数,$cnts[i] = x$ 含意为在 $[1, i]$ 范畴内质数数量为 $x$。
代码:
class Solution {static int MOD = (int)1e9+7;
static int[] cnts = new int[110];
static {List<Integer> list = new ArrayList<>();
for (int i = 2; i <= 100; i++) {
boolean ok = true;
for (int j = 2; j * j <= i; j++) {if (i % j == 0) ok = false;
}
if (ok) list.add(i);
cnts[i] = list.size();}
}
public int numPrimeArrangements(int n) {int a = cnts[n], b = n - a;
long ans = 1;
for (int i = b; i > 1; i--) ans = ans * i % MOD ;
for (int i = a; i > 1; i--) ans = ans * i % MOD ;
return (int)ans;
}
}
- 工夫复杂度:$O(n)$
- 空间复杂度:$O(C)$,其中 $C = 100$
最初
这是咱们「刷穿 LeetCode」系列文章的第 No.1175
篇,系列开始于 2021/01/01,截止于起始日 LeetCode 上共有 1916 道题目,局部是有锁题,咱们将先把所有不带锁的题目刷完。
在这个系列文章外面,除了解说解题思路以外,还会尽可能给出最为简洁的代码。如果波及通解还会相应的代码模板。
为了不便各位同学可能电脑上进行调试和提交代码,我建设了相干的仓库:https://github.com/SharingSou…。
在仓库地址里,你能够看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其余优选题解。
更多更全更热门的「口试 / 面试」相干材料可拜访排版精美的 合集新基地 🎉🎉
本文由 mdnice 多平台公布