結果
問題 | No.2709 1975 Powers |
ユーザー |
|
提出日時 | 2024-11-01 21:35:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 592 ms / 2,000 ms |
コード長 | 661 bytes |
コンパイル時間 | 421 ms |
コンパイル使用メモリ | 82,488 KB |
実行使用メモリ | 77,352 KB |
最終ジャッジ日時 | 2024-11-01 21:35:37 |
合計ジャッジ時間 | 8,397 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 25 |
ソースコード
from copy import copy from collections import defaultdict N, P, Q = map(int, input().split()) A = list(map(int, input().split())) A.sort() A_10 = [pow(10, a, P) for a in A] A_9 = [pow(9, a, P) for a in A] A_7 = [pow(7, a, P) for a in A] A_5 = [pow(5, a, P) for a in A] A_5_dict_list = [{} for _ in range(N)] di = defaultdict(int) for i in range(N - 1, -1, -1): di[A[i]] += 1 A_5_dict_list[i] = copy(di) ans = 0 for i in range(N - 3): for j in range(i + 1, N - 2): for k in range(j + 1, N - 1): for l in range(k + 1, N): if (A_10[i] + A_9[j] + A_7[k] + A_5[l]) % P == Q: ans += 1 print(ans)