結果
問題 | 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 copyfrom collections import defaultdictN, 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]] += 1A_5_dict_list[i] = copy(di)ans = 0for 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 += 1print(ans)