結果

問題 No.2709 1975 Powers
ユーザー graythunder1graythunder1
提出日時 2024-08-24 02:17:22
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 917 bytes
コンパイル時間 340 ms
コンパイル使用メモリ 82,764 KB
実行使用メモリ 629,288 KB
最終ジャッジ日時 2024-08-24 02:17:58
合計ジャッジ時間 34,659 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
52,428 KB
testcase_01 AC 34 ms
52,696 KB
testcase_02 AC 752 ms
229,540 KB
testcase_03 AC 1,223 ms
340,772 KB
testcase_04 TLE -
testcase_05 WA -
testcase_06 WA -
testcase_07 AC 237 ms
110,712 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 258 ms
114,452 KB
testcase_11 WA -
testcase_12 WA -
testcase_13 TLE -
testcase_14 AC 733 ms
222,444 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 TLE -
testcase_23 MLE -
testcase_24 WA -
testcase_25 AC 1,822 ms
494,944 KB
testcase_26 MLE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, P, Q = map(int, input().split(" "))
A = sorted(list(map(int, input().split(" "))))

def mod_pow(a, n):
    res = 1
    while n > 0:
        if n & 1:
            res = res * a % P
        a = a * a % P
        n >>= 1
    return res
    
    
# dp[i:0-4][Aj以下の数だけを使った時][あまりがkになるパターン数]
dp = [[[0 for _ in range(P)] for _ in range(N)] for _ in range(4)]
for i, x in enumerate([10, 9, 7, 5]):
    if i == 0:
        for j in range(N):
            dp[i][j][mod_pow(x, A[j])] = 1
    else:
        for j in range(i, N):
            aj_pow = mod_pow(x, A[j])
            for k in range(P):
                idx = (k + aj_pow) % P
                dp[i][j][idx] = (dp[i][j][idx] + dp[i-1][j-1][k]) % P

    for j in range(1, N):
        for k in range(P):
            dp[i][j][k] = (dp[i][j][k] + dp[i][j-1][k]) % P

ans = 0
for j in range(N):
    ans += dp[i][j][Q]
print(ans)
0