結果

問題 No.2374 ASKT Subsequences
ユーザー titia
提出日時 2023-07-11 02:54:17
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 484 ms / 2,000 ms
コード長 683 bytes
コンパイル時間 508 ms
コンパイル使用メモリ 82,644 KB
実行使用メモリ 84,732 KB
最終ジャッジ日時 2024-09-12 23:56:17
合計ジャッジ時間 7,165 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

N=int(input())
A=list(map(int,input().split()))

LIST=[[] for i in range(2001)]

for i in range(N):
    LIST[A[i]].append(i)

ANS=0

for k in range(1,2001):
    DP=[[0,0,0] for i in range(N)]
    for i in range(N):
        a=A[i]

        if 1<=a+k+10<=2000:
            for j in LIST[a+k+10]:
                if j>i:
                    DP[j][1]+=1

        if 1<=a-k<=2000:
            if DP[i][1]!=0:
                for j in LIST[a-k]:
                    if j>i:
                        DP[j][2]+=DP[i][1]

        if 1<=a+k+1<=2000:
            if DP[i][2]!=0:
                for j in LIST[a+k+1]:
                    if j>i:
                        ANS+=DP[i][2]

print(ANS)
0