#379661 (C++14) No.890 移調の限られた旋法

• 問題: No.890 移調の限られた旋法
• ユーザー: QCFium
• 提出日時: 2019-09-15 17:32:59
• 言語: C++14 (gcc 13.2.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 17 ms / 2,000 ms
• コード長: 1,235 bytes
• コンパイル時間: 194 ms
• コンパイル使用メモリ: 33,448 KB
• 実行使用メモリ: 10,860 KB
• 最終ジャッジ日時: 2023-09-21 04:15:24
• 合計ジャッジ時間: 2,133 ms
• ジャッジサーバーID （参考情報）: judge11 / judge12

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	16 ms; 10,744 KB
testcase_01	AC	16 ms; 10,780 KB
testcase_02	AC	16 ms; 10,748 KB
testcase_03	AC	16 ms; 10,748 KB
testcase_04	AC	16 ms; 10,832 KB
testcase_05	AC	16 ms; 10,720 KB
testcase_06	AC	16 ms; 10,788 KB
testcase_07	AC	16 ms; 10,860 KB
testcase_08	AC	16 ms; 10,788 KB
testcase_09	AC	16 ms; 10,672 KB
testcase_10	AC	16 ms; 10,812 KB
testcase_11	AC	16 ms; 10,772 KB
testcase_12	AC	16 ms; 10,804 KB
testcase_13	AC	17 ms; 10,812 KB
testcase_14	AC	16 ms; 10,832 KB
testcase_15	AC	16 ms; 10,796 KB
testcase_16	AC	16 ms; 10,792 KB
testcase_17	AC	16 ms; 10,776 KB
testcase_18	AC	16 ms; 10,796 KB
testcase_19	AC	16 ms; 10,848 KB
testcase_20	AC	16 ms; 10,752 KB
testcase_21	AC	16 ms; 10,724 KB
testcase_22	AC	16 ms; 10,748 KB
testcase_23	AC	16 ms; 10,848 KB
testcase_24	AC	17 ms; 10,744 KB
testcase_25	AC	17 ms; 10,748 KB
testcase_26	AC	16 ms; 10,796 KB
testcase_27	AC	17 ms; 10,800 KB
testcase_28	AC	16 ms; 10,808 KB
testcase_29	AC	16 ms; 10,748 KB
testcase_30	AC	16 ms; 10,804 KB
testcase_31	AC	16 ms; 10,732 KB
testcase_32	AC	16 ms; 10,792 KB
testcase_33	AC	16 ms; 10,820 KB
testcase_34	AC	16 ms; 10,804 KB

ソースコード

#include <stdio.h>
#include <stdint.h>
#include <assert.h>
#include <string.h>

int ri() {
	int n;
	scanf("%d", &n);
	return n;
}
#define MOD 1000000007
int main() {
	int n, k;
	scanf("%d%d", &n, &k);
	assert(1 <= k && k <= n && n <= 1000000);
	static int fact[1000001];
	fact[0] = 1;
	for (int i = 1; i <= 1000000; i++) fact[i] = (int64_t) fact[i - 1] * i % MOD;
	int cur = fact[1000000];
	static int inv[1000001];
	inv[1000000] = 1;
	for (int i = 0; i < 30; i++) {
		if ((MOD - 2) >> i & 1) inv[1000000] = (int64_t) inv[1000000] * cur % MOD;
		cur = (int64_t) cur * cur % MOD;
	}
	for (int i = 1000000; i--; ) inv[i] = (int64_t) inv[i + 1] * (i + 1) % MOD;
	int a = n, b = k;
	while (a && b) {
		if (a > b) a %= b;
		else b %= a;
	}
	int p[20], n_p = 0, t = a + b;
	for (int i = 2; i < t; i++) if (t % i == 0) {
		p[n_p++] = i;
		while (t % i == 0) t /= i;
	}
	if (t > 1) p[n_p++] = t;
	int res = 0;
	for (int i = 1; i < 1 << n_p; i++) {
		int cur = 1;
		for (int j = 0; j < n_p; j++) if (i >> j & 1) cur *= p[j];
		int add = (int64_t) fact[n / cur] * inv[k / cur] % MOD * inv[(n - k) / cur] % MOD;
		if (__builtin_popcount(i) & 1) res = (res + add) % MOD;
		else res = (res - add + MOD) % MOD;
	}
	printf("%d\n", res);
	return 0;
}