結果

問題 No.2374 ASKT Subsequences
コンテスト
ユーザー Kanten4205
提出日時 2023-06-13 08:06:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,620 ms / 2,000 ms
コード長 695 bytes
コンパイル時間 238 ms
コンパイル使用メモリ 82,376 KB
実行使用メモリ 230,980 KB
最終ジャッジ日時 2024-06-29 22:31:27
合計ジャッジ時間 14,038 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
A = list(map(int, input().split()))
Max = max(A)
ans = 0
for k in range(1,Max + 1):
    dp = [[0 for i in range(4)] for j in range(N)]
    C = [[0 for i in range(4)] for j in range(Max + 1)]
    for i in range(N):
        dp[i][3] = 1
    C[A[N - 1]][3] = 1
    for i in reversed(range(0, N - 1)):
        if 0 <= A[i] + k + 1 and A[i] + k + 1 <= Max: dp[i][2] = C[A[i] + k + 1][3]
        if 0 <= A[i] - k and A[i] - k <= Max: dp[i][1] = C[A[i] - k][2]
        if 0 <= A[i] + 10 + k and A[i] + 10 + k <= Max: dp[i][0] = C[A[i] + 10 + k][1]
        for j in range(4):
            C[A[i]][j] = C[A[i]][j] + dp[i][j]
    for i in range(N):
        ans = ans + dp[i][0]
print(ans)
0