結果
問題 | No.1035 Color Box |
ユーザー |
|
提出日時 | 2021-07-02 23:03:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 230 ms / 2,000 ms |
コード長 | 861 bytes |
コンパイル時間 | 163 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 112,108 KB |
最終ジャッジ日時 | 2024-06-29 12:55:19 |
合計ジャッジ時間 | 7,049 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
#!/usr/bin/env python3# from typing import *import sysimport ioimport mathimport collectionsimport decimalimport itertoolsimport bisectimport heapqdef input():return sys.stdin.readline()[:-1]# sys.setrecursionlimit(1000000)# _INPUT = """# paste here...# """# sys.stdin = io.StringIO(_INPUT)MOD = 1000000007def comb(n, r):r = min(r, n - r)return (fact[n] * factinv[r] * factinv[n-r]) % MODfact = [1, 1] # fact[n] = (n! mod p)factinv = [1, 1] # factinv[n] = ((n!)^(-1) mod p)inv = [0, 1] # factinv 計算用N, M = map(int, input().split())for i in range(2, M + 1):fact.append((fact[-1] * i) % MOD)inv.append((-inv[MOD % i] * (MOD // i)) % MOD)factinv.append((factinv[-1] * inv[-1]) % MOD)ans = 0for k in range(M):ans += (-1)**k * comb(M, k) * pow(M-k, N, MOD)ans %= MODprint(ans)