結果
問題 | No.1035 Color Box |
ユーザー |
|
提出日時 | 2024-11-01 21:09:59 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 101 ms / 2,000 ms |
コード長 | 433 bytes |
コンパイル時間 | 356 ms |
コンパイル使用メモリ | 82,284 KB |
実行使用メモリ | 75,404 KB |
最終ジャッジ日時 | 2024-11-01 21:10:04 |
合計ジャッジ時間 | 4,065 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
MOD = 10**9+7N,M = map(int,input().split())A = list(range(M+1))A[0]=A[1]=1for i in range(2,M+1):A[i]=(i*A[i-1])%MODB = [0]*(M+1)B[0]=B[1]=1B[M] = pow(A[M],MOD-2,MOD)for i in range(M-1,1,-1):B[i] = (B[i+1]*(i+1))%MODdef comb(n,k):if k>n or k<0:return 0if k==0 or k==n:return 1return (A[n]*B[k]*B[n-k])%MODans = 0for k in range(M):ans = (ans+(-1)**(k%2)*comb(M,k)*pow(M-k,N,MOD))%MODprint(ans)