結果

問題 No.2709 1975 Powers
ユーザー Nikkuniku029
提出日時 2024-03-31 17:30:32
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 226 ms / 2,000 ms
コード長 635 bytes
コンパイル時間 353 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 89,016 KB
最終ジャッジ日時 2024-09-30 21:11:37
合計ジャッジ時間 4,265 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, P, Q = map(int, input().split())
A = list(map(int, input().split()))
A.sort()
from collections import defaultdict
from bisect import bisect_right

five = defaultdict(list)
seven = defaultdict(int)
nine = defaultdict(int)
ten = defaultdict(int)
for a in A:
    seven[a] = pow(7, a, P)
    nine[a] = pow(9, a, P)
    ten[a] = pow(10, a, P)
    five[pow(5, a, P)].append(a)
ans = 0
for a in range(N):
    for b in range(a + 1, N):
        for c in range(b + 1, N):
            x = (ten[A[a]] + nine[A[b]] + seven[A[c]]) % P
            y = len(five[(Q - x) % P]) - bisect_right(five[(Q - x) % P], A[c])
            ans += y
print(ans)
0