結果

問題 No.2709 1975 Powers
ユーザー miya145592miya145592
提出日時 2024-03-31 14:17:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,042 ms / 2,000 ms
コード長 612 bytes
コンパイル時間 262 ms
コンパイル使用メモリ 82,140 KB
実行使用メモリ 260,436 KB
最終ジャッジ日時 2024-09-30 19:18:25
合計ジャッジ時間 14,645 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
N, P, Q = map(int, input().split())
A = list(map(int, input().split()))
A.sort()
dp = [[0 for _ in range(P)] for _ in range(N+1)]
for i in range(N):
    q = pow(5, A[i], P)
    dp[i+1] = dp[i][:]
    dp[i+1][q]+=1
ans = 0
for i in range(N):
    a = A[i]
    tmpa = pow(10, a, P)
    for j in range(i+1, N):
        b = A[j]
        tmpb = pow(9, b, P)
        for k in range(j+1, N):
            c = A[k]
            tmp = (tmpa+tmpb)%P+pow(7, c, P)
            tmp %= P
            tmp = (Q-tmp)%P
            cnt = dp[N][tmp]-dp[k+1][tmp]
            ans += cnt
print(ans)
0