#469314 (Python3) No.1035 Color Box

• 問題: No.1035 Color Box
• ユーザー: anagohirame
• 提出日時: 2020-04-24 21:47:14
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: AC
• 実行時間: 469 ms / 2,000 ms
• コード長: 697 bytes
• コンパイル時間: 77 ms
• コンパイル使用メモリ: 12,672 KB
• 実行使用メモリ: 18,560 KB
• 最終ジャッジ日時: 2024-10-15 02:40:53
• 合計ジャッジ時間: 9,305 ms
• ジャッジサーバーID （参考情報）: judge4 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	469 ms; 18,560 KB
testcase_01	AC	161 ms; 18,560 KB
testcase_02	AC	160 ms; 18,432 KB
testcase_03	AC	184 ms; 18,432 KB
testcase_04	AC	175 ms; 18,432 KB
testcase_05	AC	163 ms; 18,432 KB
testcase_06	AC	153 ms; 18,560 KB
testcase_07	AC	154 ms; 18,432 KB
testcase_08	AC	405 ms; 18,432 KB
testcase_09	AC	273 ms; 18,432 KB
testcase_10	AC	159 ms; 18,432 KB
testcase_11	AC	152 ms; 18,432 KB
testcase_12	AC	149 ms; 18,304 KB
testcase_13	AC	373 ms; 18,432 KB
testcase_14	AC	414 ms; 18,304 KB
testcase_15	AC	465 ms; 18,432 KB
testcase_16	AC	159 ms; 18,560 KB
testcase_17	AC	156 ms; 18,560 KB
testcase_18	AC	151 ms; 18,432 KB
testcase_19	AC	465 ms; 18,432 KB
testcase_20	AC	160 ms; 18,432 KB
testcase_21	AC	161 ms; 18,560 KB
testcase_22	AC	159 ms; 18,304 KB
testcase_23	AC	151 ms; 18,560 KB
testcase_24	AC	148 ms; 18,432 KB
testcase_25	AC	152 ms; 18,560 KB
testcase_26	AC	161 ms; 18,304 KB
testcase_27	AC	148 ms; 18,560 KB
testcase_28	AC	147 ms; 18,560 KB
testcase_29	AC	148 ms; 18,560 KB
testcase_30	AC	146 ms; 18,432 KB
testcase_31	AC	152 ms; 18,560 KB
testcase_32	AC	139 ms; 18,560 KB
testcase_33	AC	149 ms; 18,304 KB
testcase_34	AC	143 ms; 18,560 KB
testcase_35	AC	145 ms; 18,432 KB
testcase_36	AC	149 ms; 18,560 KB
testcase_37	AC	157 ms; 18,560 KB

ソースコード

MOD = 10**9 + 7
list_size = 100001

f_list = [1] * list_size
f_r_list = [1] * list_size

for i in range(list_size - 1):
	f_list[i + 1] = int((f_list[i] * (i + 2)) % MOD)

f_r_list[-1] = pow(f_list[-1], MOD - 2, MOD)

for i in range(2, list_size + 1):
	f_r_list[-i] = int((f_r_list[-i + 1] * (list_size + 2 - i)) % MOD)

def comb(n, r):
	if n < r:
		return 0
	elif n == 0 or r == 0 or n == r:
		return 1
	else:
		return (((f_list[n - 1] * f_r_list[n - r - 1]) % MOD) * f_r_list[r - 1]) % MOD 

n, m = map(int, input().split())
ans = 0
for i in range(m, 0, -1):
	if (m%2) == (i%2):
		ans += pow(i, n, MOD) * comb(m, i)
		ans %= MOD
	else:
		ans -= pow(i, n, MOD) * comb(m, i)
		ans %= MOD
print(ans)