結果
問題 | No.2423 Merge Stones |
ユーザー |
|
提出日時 | 2023-07-15 18:00:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,319 ms / 4,000 ms |
コード長 | 878 bytes |
コンパイル時間 | 150 ms |
コンパイル使用メモリ | 82,512 KB |
実行使用メモリ | 79,368 KB |
最終ジャッジ日時 | 2024-09-17 04:03:36 |
合計ジャッジ時間 | 79,797 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 72 |
ソースコード
n, k = map(int, input().split()) a = list(map(int, input().split())) c = list(map(lambda x: int(x)-1, input().split())) dp = [[0] * 2 * n for _ in range(2 * n)] for i in range(n): dp[i][i] = 1 << c[i] dp[i + n][i + n] = 1 << c[i] for len in range(2, n + 1): for l in range(2 * n - len + 1): r = l + len - 1 for i in range(len - 1): m = l + i bitl, bitr = dp[l][m], dp[m + 1][r] bitlr = dp[l][r] for p in range(k + 1): if dp[l][m] == 0 or dp[m + 1][r] == 0: continue bitlr |= ((bitl << p) & bitr) bitlr |= ((bitl >> p) & bitr) bitlr |= (bitl & (bitr << p)) bitlr |= (bitl & (bitr >> p)) dp[l][r] = bitlr s = [0] for i in range(2 * n): s.append(s[i] + a[i % n]) ans = 0 for i in range(n): for j in range(i, i + n): if dp[i][j]: ans = max(ans, s[j + 1] - s[i]) print(ans)