結果
問題 | No.1368 サイクルの中に眠る門松列 |
ユーザー |
![]() |
提出日時 | 2021-01-29 22:34:28 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 860 ms / 2,000 ms |
コード長 | 1,217 bytes |
コンパイル時間 | 217 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 32,388 KB |
最終ジャッジ日時 | 2024-06-27 08:52:39 |
合計ジャッジ時間 | 7,034 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 15 |
ソースコード
def is_kadomatsu(a, b, c): if a == b or b == c or a == c: return False return min([a, b, c]) == b or max([a, b, c]) == b def solve(): dp = [0] * (N + 1) for i in range(N + 1): if i != 0: dp[i] = max(dp[i], dp[i - 1]) if i <= N - 3: if is_kadomatsu(A[i], A[i + 1], A[i + 2]): dp[i + 3] = dp[i] + A[i] result = dp[N] if is_kadomatsu(A[N - 1], A[0], A[1]): dp = [0] * (N + 1) dp[2] = A[N - 1] for i in range(2, N + 1): dp[i] = max(dp[i], dp[i - 1]) if i <= N - 4: if is_kadomatsu(A[i], A[i + 1], A[i + 2]): dp[i + 3] = dp[i] + A[i] result = max(result, dp[N]) if is_kadomatsu(A[N - 2], A[N - 1], A[0]): dp = [0] * (N + 1) dp[1] = A[N - 2] for i in range(1, N + 1): dp[i] = max(dp[i], dp[i - 1]) if i <= N - 5: if is_kadomatsu(A[i], A[i + 1], A[i + 2]): dp[i + 3] = dp[i] + A[i] result = max(result, dp[N]) return result T = int(input()) for _ in range(T): N = int(input()) A = list(map(int, input().split())) print(solve())