結果
問題 |
No.2374 ASKT Subsequences
|
ユーザー |
|
提出日時 | 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 |
ソースコード
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)