結果
問題 |
No.2374 ASKT Subsequences
|
ユーザー |
|
提出日時 | 2023-06-13 19:20:59 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,787 ms / 2,000 ms |
コード長 | 690 bytes |
コンパイル時間 | 165 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 228,368 KB |
最終ジャッジ日時 | 2024-06-29 22:31:11 |
合計ジャッジ時間 | 13,512 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
N = int(input()) A = list(map(int, input().split())) Max = max(A);Min = min(A) ans = 0 for k in range(1,Max - Min + 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)] dp[0][0] = 1 C[A[0]][0] = 1 for i in range(1, N): dp[i][0] = 1 if 1 <= A[i] - k - 10 and A[i] - k - 10 <= Max: dp[i][1] = C[A[i] - k - 10][0] if 1 <= A[i] + k and A[i] + k <= Max: dp[i][2] = C[A[i] + k][1] if 1 <= A[i] - 1 - k and A[i] - 1 - k <= Max: dp[i][3] = C[A[i] - 1 - k][2] 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][3] print(ans)