結果
問題 | No.2374 ASKT Subsequences |
ユーザー |
![]() |
提出日時 | 2023-07-07 22:40:54 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 254 ms / 2,000 ms |
コード長 | 1,066 bytes |
コンパイル時間 | 410 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 108,928 KB |
最終ジャッジ日時 | 2024-07-21 18:56:13 |
合計ジャッジ時間 | 5,039 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
# 二重ループ可能 # a2とa3を決めれば、k, a1, a4も決まる # a2とa3を決めたときにa2までのa1の個数かけるa3後のa4の個数 # インデックス番号までにどの数字が何個出ていたかカウントする N = int(input()) A = list(map(int, input().split())) A_max = 2001 # 2001にすること count = [] old = [0]*A_max count.append(old) for i in range(N): new = [] for j in range(A_max): new.append(old[j]) new[A[i]] += 1 count.append(new) old = new ans = 0 for a2_idx in range(1, N-2): for a3_idx in range(a2_idx+1, N-1): #print('a2_idx', a2_idx, 'a3_idx', a3_idx) a2 = A[a2_idx] k = a2 - A[a3_idx] if k <= 0: continue a1 = a2-(k+10) a4 = a2+1 if a1 <= 0 or A_max <= a1: continue if a4 <= 0 or A_max <= a4: continue add = count[a2_idx][a1]*(count[N][a4]-count[a3_idx+1][a4]) #print('a2', a2, 'k', k, 'add', add) ans += add print(ans)