#568189 (C) No.1259 スイッチ

提出ソース

結果

問題	No.1259 スイッチ
ユーザー	ygussany
提出日時	2020-10-16 23:51:35
言語	C (gcc 12.3.0)
結果	AC
実行時間	41 ms / 2,000 ms
コード長	1,119 bytes
コンパイル時間	1,215 ms
コンパイル使用メモリ	31,104 KB
実行使用メモリ	25,344 KB
最終ジャッジ日時	2024-07-21 01:04:39
合計ジャッジ時間	3,645 ms
ジャッジサーバーID （参考情報）	judge3 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	0 ms 5,248 KB
testcase_01	AC	1 ms 5,376 KB
testcase_02	AC	1 ms 5,376 KB
testcase_03	AC	1 ms 5,376 KB
testcase_04	AC	1 ms 5,376 KB
testcase_05	AC	1 ms 5,376 KB
testcase_06	AC	1 ms 5,376 KB
testcase_07	AC	1 ms 5,376 KB
testcase_08	AC	1 ms 5,376 KB
testcase_09	AC	1 ms 5,376 KB
testcase_10	AC	26 ms 15,872 KB
testcase_11	AC	39 ms 23,680 KB
testcase_12	AC	24 ms 15,104 KB
testcase_13	AC	14 ms 9,728 KB
testcase_14	AC	20 ms 13,056 KB
testcase_15	AC	14 ms 9,344 KB
testcase_16	AC	32 ms 19,968 KB
testcase_17	AC	19 ms 12,672 KB
testcase_18	AC	37 ms 22,400 KB
testcase_19	AC	37 ms 22,272 KB
testcase_20	AC	25 ms 16,000 KB
testcase_21	AC	39 ms 24,320 KB
testcase_22	AC	12 ms 8,576 KB
testcase_23	AC	36 ms 21,888 KB
testcase_24	AC	26 ms 17,024 KB
testcase_25	AC	26 ms 16,512 KB
testcase_26	AC	40 ms 24,704 KB
testcase_27	AC	33 ms 20,352 KB
testcase_28	AC	26 ms 16,640 KB
testcase_29	AC	33 ms 20,224 KB
testcase_30	AC	18 ms 13,184 KB
testcase_31	AC	26 ms 18,304 KB
testcase_32	AC	12 ms 8,704 KB
testcase_33	AC	36 ms 24,832 KB
testcase_34	AC	20 ms 14,208 KB
testcase_35	AC	28 ms 19,456 KB
testcase_36	AC	26 ms 17,920 KB
testcase_37	AC	24 ms 17,024 KB
testcase_38	AC	22 ms 15,744 KB
testcase_39	AC	34 ms 23,680 KB
testcase_40	AC	34 ms 23,936 KB
testcase_41	AC	28 ms 19,968 KB
testcase_42	AC	34 ms 24,320 KB
testcase_43	AC	21 ms 15,104 KB
testcase_44	AC	34 ms 23,296 KB
testcase_45	AC	36 ms 25,216 KB
testcase_46	AC	19 ms 14,464 KB
testcase_47	AC	35 ms 25,088 KB
testcase_48	AC	17 ms 12,928 KB
testcase_49	AC	34 ms 24,320 KB
testcase_50	AC	22 ms 14,464 KB
testcase_51	AC	12 ms 8,576 KB
testcase_52	AC	37 ms 23,808 KB
testcase_53	AC	17 ms 11,392 KB
testcase_54	AC	38 ms 23,424 KB
testcase_55	AC	12 ms 8,320 KB
testcase_56	AC	19 ms 12,800 KB
testcase_57	AC	35 ms 21,760 KB
testcase_58	AC	24 ms 15,488 KB
testcase_59	AC	29 ms 18,688 KB
testcase_60	AC	41 ms 25,344 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <stdio.h>

const int Mod = 1000000007;

long long div_mod(long long x, long long y, long long z)
{
	if (x % y == 0) return x / y;
	else return (div_mod((1 + x / y) * y - x, (z % y), y) * z + x) / y;
}

int main()
{
	int N, M;
	long long K;
	scanf("%d %lld %d", &N, &K, &M);
	
	int i;
	long long pow[1000001], fact[1000001], fact_inv[1000001], ans = 0;
	for (i = 1, pow[0] = 1; i <= N; i++) pow[i] = pow[i-1] * N % Mod;
	for (i = 1, fact[0] = 1; i <= N; i++) fact[i] = fact[i-1] * i % Mod;
	for (i = N - 1, fact_inv[N] = div_mod(1, fact[N], Mod); i >= 0; i--) fact_inv[i] = fact_inv[i+1] * (i + 1) % Mod;
	
	if (M == 1) {
		for (i = 1; i <= N; i++) {
			if (K % i != 0) continue;
			ans += fact[N-1] * fact_inv[N-i] % Mod * pow[N-i] % Mod;
		}
	} else {
		if (K < N) ans += fact[N-2] * fact_inv[N-K-1] % Mod * pow[N-K] % Mod;
		for (i = 2; i <= N; i++) {
			if (i > K) break;
			if (K % i != 0) ans += fact[N-2] * fact_inv[N-i] % Mod * i % Mod * pow[N-i] % Mod;
			else ans += fact[N-2] * fact_inv[N-i] % Mod * (i - 1) % Mod * pow[N-i] % Mod;
		}
	}
	
	printf("%lld\n", ans % Mod);
	fflush(stdout);
	return 0;
}

yukicoder

結果

テストケース

ソースコード