結果
問題 |
No.2709 1975 Powers
|
ユーザー |
|
提出日時 | 2024-04-18 10:26:04 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 564 bytes |
コンパイル時間 | 474 ms |
コンパイル使用メモリ | 82,380 KB |
実行使用メモリ | 157,396 KB |
最終ジャッジ日時 | 2024-10-10 03:39:20 |
合計ジャッジ時間 | 26,736 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 7 WA * 14 TLE * 4 |
ソースコード
from collections import defaultdict import copy N, P, Q=map(int, input().split()) A=list(map(int, input().split())) stset=[None for i in range(N)] for i in range(len(A)-1, -1, -1): if i==len(A)-1: obj=defaultdict(int) obj[pow(5, A[i], P)]+=1 stset[i]=obj continue obj=copy.deepcopy(stset[i+1]) obj[pow(5, A[i], P)]+=1 stset[i]=obj ans=0 for a in range(N-4+1): for b in range(a+1, N-3+1): for c in range(b+1, N-2+1): v=pow(10, A[a], P)+pow(9, A[b], P)+pow(7, A[c], P) target=(Q-v)%P ans+=stset[c+1][target] print(ans)