AtCoder Beginner Contest 227G题Divisors of Binomial Coefficient（数论）

题目链接

大概意思是求组合数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;
}