結果
問題 |
No.2374 ASKT Subsequences
|
ユーザー |
|
提出日時 | 2023-06-10 11:32:09 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,552 ms / 2,000 ms |
コード長 | 685 bytes |
コンパイル時間 | 157 ms |
コンパイル使用メモリ | 82,400 KB |
実行使用メモリ | 229,388 KB |
最終ジャッジ日時 | 2024-06-29 22:30:14 |
合計ジャッジ時間 | 13,924 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
n = int(input()) A = list(map(int, input().split())) Max = max(A) ans = 0 for k in range(1,Max + 1): B = [[0 for i in range(4)] for j in range(n)] C = [[0 for i in range(4)] for j in range(Max + 1)] B[n - 1][3] = 1 C[A[n - 1]][3] = 1 for i in reversed(range(0, n - 1)): B[i][3] = 1 if 0 <= A[i] + k + 1 and A[i] + k + 1 <= Max: B[i][2] = C[A[i] + k + 1][3] if 0 <= A[i] - k and A[i] - k <= Max: B[i][1] = C[A[i] - k][2] if 0 <= A[i] + 10 + k and A[i] + 10 + k <= Max: B[i][0] = C[A[i] + 10 + k][1] for j in range(4): C[A[i]][j] = C[A[i]][j] + B[i][j] for i in range(n): ans = ans + B[i][0] print(ans)