結果

問題 No.1025 Modular Equation
ユーザー chocorusk
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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])
0