結果

問題 No.1025 Modular Equation
ユーザー chocoruskchocorusk
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
53,672 KB
testcase_01 AC 36 ms
52,488 KB
testcase_02 AC 52 ms
63,000 KB
testcase_03 AC 50 ms
63,828 KB
testcase_04 AC 50 ms
64,660 KB
testcase_05 AC 50 ms
64,728 KB
testcase_06 AC 50 ms
64,732 KB
testcase_07 AC 54 ms
67,376 KB
testcase_08 AC 51 ms
63,656 KB
testcase_09 AC 1,339 ms
73,328 KB
testcase_10 AC 1,390 ms
72,140 KB
testcase_11 AC 1,303 ms
74,260 KB
testcase_12 AC 1,419 ms
74,480 KB
testcase_13 AC 1,231 ms
73,716 KB
testcase_14 AC 1,305 ms
74,648 KB
testcase_15 AC 1,305 ms
78,916 KB
testcase_16 AC 1,247 ms
78,684 KB
testcase_17 AC 1,287 ms
78,808 KB
testcase_18 AC 1,208 ms
78,580 KB
testcase_19 AC 2,032 ms
78,504 KB
testcase_20 AC 2,065 ms
78,756 KB
testcase_21 AC 1,690 ms
78,840 KB
testcase_22 AC 1,674 ms
78,856 KB
testcase_23 AC 1,194 ms
71,648 KB
testcase_24 AC 1,497 ms
73,500 KB
testcase_25 AC 1,430 ms
74,300 KB
testcase_26 AC 1,391 ms
73,988 KB
testcase_27 AC 1,335 ms
79,392 KB
testcase_28 AC 1,367 ms
78,940 KB
testcase_29 AC 1,334 ms
79,168 KB
testcase_30 AC 1,784 ms
79,264 KB
testcase_31 AC 2,174 ms
78,940 KB
testcase_32 AC 1,307 ms
71,572 KB
testcase_33 AC 38 ms
55,528 KB
testcase_34 AC 973 ms
78,576 KB
権限があれば一括ダウンロードができます

ソースコード

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