結果
問題 | No.2396 等差二項展開 |
ユーザー | tassei903 |
提出日時 | 2023-07-28 22:43:44 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 3,252 ms / 6,000 ms |
コード長 | 1,097 bytes |
コンパイル時間 | 313 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 77,056 KB |
最終ジャッジ日時 | 2024-10-06 20:19:40 |
合計ジャッジ時間 | 23,494 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 31 |
ソースコード
import sys input = lambda :sys.stdin.readline()[:-1] ni = lambda :int(input()) na = lambda :list(map(int,input().split())) yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES") no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO") ####################################################################### n,m,l,k,b = na() if l == 1: print(pow(1+m, n, b)) exit() a = [0] * l a[0] = 1 % b a[1] = 1 % b f = [0] * l f[0] = 1 #print(f, a) while n: if n % 2: nf = [0] * l for i in range(l): for j in range(l): if i+j >= l: nf[i+j-l] += f[i] * a[j] % b * m % b else: nf[i+j] += f[i] * a[j] % b nf[(i+j)%l] %= b f = nf na = [0] * l for i in range(l): for j in range(l): if i+j >= l: na[i+j-l] += a[i] * a[j] % b * m % b else: na[i+j] += a[i] * a[j] % b na[(i+j)%l] %= b a = na #print(f, a) n//=2 #print(f) print(f[k] % b)