結果
問題 | No.1035 Color Box |
ユーザー |
![]() |
提出日時 | 2020-04-24 21:29:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 94 ms / 2,000 ms |
コード長 | 851 bytes |
コンパイル時間 | 183 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 65,024 KB |
最終ジャッジ日時 | 2024-10-15 02:23:14 |
合計ジャッジ時間 | 3,509 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
mod = 1000000007eps = 10**-9def main():import sysinput = sys.stdin.buffer.readline# comb initnmax = 2 * 10 ** 5 + 10 # change herefac = [0] * nmaxfinv = [0] * nmaxinv = [0] * nmaxfac[0] = 1fac[1] = 1finv[0] = 1finv[1] = 1inv[1] = 1for i in range(2, nmax):fac[i] = fac[i - 1] * i % modinv[i] = mod - inv[mod % i] * (mod // i) % modfinv[i] = finv[i - 1] * inv[i] % moddef comb(n, r):if n < r:return 0else:return (fac[n] * ((finv[r] * finv[n - r]) % mod)) % modN, M = map(int, input().split())ans = pow(M, N, mod)flg = -1for i in range(M-1, 0, -1):ans = (ans + flg * comb(M, i) * pow(i, N, mod)) % modflg *= -1print(ans)if __name__ == '__main__':main()