結果

問題 No.1025 Modular Equation
ユーザー chocoruskchocorusk
提出日時 2020-04-08 13:52:41
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,400 ms / 5,000 ms
コード長 798 bytes
コンパイル時間 1,772 ms
コンパイル使用メモリ 86,572 KB
実行使用メモリ 79,908 KB
最終ジャッジ日時 2023-09-25 14:06:05
合計ジャッジ時間 44,197 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge12
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 71 ms
71,508 KB
testcase_01 AC 75 ms
71,044 KB
testcase_02 AC 92 ms
76,384 KB
testcase_03 AC 89 ms
76,184 KB
testcase_04 AC 88 ms
76,536 KB
testcase_05 AC 87 ms
76,580 KB
testcase_06 AC 87 ms
76,696 KB
testcase_07 AC 90 ms
76,568 KB
testcase_08 AC 86 ms
76,624 KB
testcase_09 AC 1,481 ms
79,328 KB
testcase_10 AC 1,541 ms
79,492 KB
testcase_11 AC 1,423 ms
79,656 KB
testcase_12 AC 1,542 ms
79,808 KB
testcase_13 AC 1,369 ms
79,480 KB
testcase_14 AC 1,461 ms
79,672 KB
testcase_15 AC 1,457 ms
79,812 KB
testcase_16 AC 1,380 ms
79,228 KB
testcase_17 AC 1,389 ms
79,880 KB
testcase_18 AC 1,334 ms
79,604 KB
testcase_19 AC 2,190 ms
79,756 KB
testcase_20 AC 2,231 ms
79,520 KB
testcase_21 AC 1,844 ms
79,816 KB
testcase_22 AC 1,771 ms
79,740 KB
testcase_23 AC 1,312 ms
79,380 KB
testcase_24 AC 1,703 ms
79,828 KB
testcase_25 AC 1,567 ms
79,908 KB
testcase_26 AC 1,562 ms
79,536 KB
testcase_27 AC 1,480 ms
79,892 KB
testcase_28 AC 1,493 ms
79,816 KB
testcase_29 AC 1,478 ms
79,808 KB
testcase_30 AC 1,965 ms
79,804 KB
testcase_31 AC 2,400 ms
79,808 KB
testcase_32 AC 1,398 ms
79,624 KB
testcase_33 AC 74 ms
71,544 KB
testcase_34 AC 1,095 ms
79,348 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