結果
問題 | No.1025 Modular Equation |
ユーザー |
![]() |
提出日時 | 2020-04-08 13:52:41 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,174 ms / 5,000 ms |
コード長 | 798 bytes |
コンパイル時間 | 158 ms |
コンパイル使用メモリ | 82,424 KB |
実行使用メモリ | 79,392 KB |
最終ジャッジ日時 | 2024-07-18 11:05:33 |
合計ジャッジ時間 | 37,905 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 32 |
ソースコード
import math p, n, k, b=map(int, input().split()) a=list(map(int, input().split())) r=1 for i in range(1, p): dame=False pp=1 for j in range(p-2): pp*=i pp%=p if pp==1: dame=True break if not dame: r=i break q=[1]*(p-1) k=math.gcd(k, p-1) for i in range(1, p-1): q[i]=q[i-1]*r%p MOD=10**9+7 dp=[0]*p tmp=[0]*p dp[0]=1 for ai in a: for i in range(p): tmp[i]=dp[i] for j in range(0, p-1, k): tmp[0]+=dp[(-ai*q[j])%p]*k tmp[0]%=MOD for i in range(k): for j in range(0, p-1, k): tmp[q[i]]+=dp[(q[i]-ai*q[j])%p]*k tmp[q[i]]%=MOD for j in range(i, p-1, k): tmp[q[j]]=tmp[q[i]] for i in range(p): dp[i]=tmp[i] print(dp[b])