結果

問題 No.2374 ASKT Subsequences
コンテスト
ユーザー FromBooska
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 二重ループ可能
# 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)
0