結果
| 問題 | 
                            No.1035 Color Box
                             | 
                    
| コンテスト | |
| ユーザー | 
                             tpyneriver
                         | 
                    
| 提出日時 | 2020-04-24 21:43:16 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                AC
                                 
                             
                            
                         | 
                    
| 実行時間 | 202 ms / 2,000 ms | 
| コード長 | 624 bytes | 
| コンパイル時間 | 145 ms | 
| コンパイル使用メモリ | 82,028 KB | 
| 実行使用メモリ | 121,124 KB | 
| 最終ジャッジ日時 | 2024-10-15 02:38:22 | 
| 合計ジャッジ時間 | 6,354 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge3 / judge2 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 2 | 
| other | AC * 36 | 
ソースコード
import sys
readline = sys.stdin.readline
MOD = 10**9+7
def frac(limit):
    frac = [1]*limit
    for i in range(2,limit):
        frac[i] = i * frac[i-1]%MOD
    fraci = [None]*limit
    fraci[-1] = pow(frac[-1], MOD -2, MOD)
    for i in range(-2, -limit-1, -1):
        fraci[i] = fraci[i+1] * (limit + i + 1) % MOD
    return frac, fraci
def comb(a, b):
    if not a >= b >= 0:
        return 0
    return frac[a]*fraci[b]*fraci[a-b]%MOD
frac, fraci = frac(1341398)
N, M = map(int, readline().split())
ans = pow(M, N, MOD)
for k in range(1, M):
    ans = (ans + (-1)**k * comb(M, k) * pow(M-k, N, MOD))
print(ans % MOD)
            
            
            
        
            
tpyneriver