题目链接
大概意思是求组合数C(N,K)的因子个数
由于K<=1e6
显然组合数拆分后可以得到n*(n-1)***(n-k+1)/k!
因子个数可以由每种质因子的幂次计算得到。
把式子上下消除后按质因子幂次归纳,可以得到一个新的式子
此时我们只需要处理1–K的质数并计算出第i个质数在k!中出现了多少次,在分子又出现了多少次
然后计算答案,最终统计下分子上方仍然不为1的(说明是大质数)答案乘2
#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10, P = 998244353;
int p[N], fac[N], pcnt;
long long cnt[N], haji;
long long fuck[N];
long long n, k, g, q, ans;
int main () {
scanf ("%lld%lld", &n, &k);
if (k > n - k) k = n - k;
g = max ((long long)sqrt (n), k);
// 计算n! / (n - k)!相关的质因子。
for (long long i = n - k + 1; i <= n; ++i)
fuck[i - (n - k)] = i;
for (int i = 2; i <= g; ++i) {
if (!fac[i]) {
fac[i] = p[++pcnt] = i;
q = i;
while (q <= n) {
cnt[pcnt] += n / q - (n - k) / q - k / q;
q = q * i;
}
haji = (n - k + 1) % i ? i - ((n - k + 1) % i) + n - k + 1 : n - k + 1;
for (; haji <= n; haji += i)
while (fuck[haji - (n - k)] % i == 0)
fuck[haji - (n - k)] /= i;
}
for (int j = 1; j <= pcnt; ++j) {
if (i * p[j] > g || fac[i] < p[j]) break;
fac[i * p[j]] = p[j];
}
}
for (long long i = n - k + 1; i <= n; ++i)
if (fuck[i - (n - k)] != 1) cnt[++pcnt] = 1, p[pcnt] = fuck[i - (n - k)];
ans = cnt[1] + 1;
for (int i = 2; i <= pcnt; ++i)
ans = 1ll * ans * (cnt[i] + 1) % P;
printf ("%d\n", ans);
return 0;
}