結果
問題 | No.2423 Merge Stones |
ユーザー |
![]() |
提出日時 | 2023-07-15 17:00:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,768 ms / 4,000 ms |
コード長 | 780 bytes |
コンパイル時間 | 206 ms |
コンパイル使用メモリ | 82,468 KB |
実行使用メモリ | 77,388 KB |
最終ジャッジ日時 | 2024-09-17 04:05:47 |
合計ジャッジ時間 | 125,406 ms |
ジャッジサーバーID (参考情報) |
judge6 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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] * n for _ in range(n)] for i in range(n): dp[i][i] = 1 << c[i] for i in range(1, n): for l in range(n): r = (l + i) % n for j in range(i): m = (l + j) % n for p in range(k+1): dp[l][r] |= ((dp[l][m] << p) & dp[(m+1)%n][r]) dp[l][r] |= ((dp[l][m] >> p) & dp[(m+1)%n][r]) dp[l][r] |= (dp[l][m] & (dp[(m+1)%n][r] << p)) dp[l][r] |= (dp[l][m] & (dp[(m+1)%n][r] >> p)) s = [0] for i in range(n): s.append(s[i]+a[i]) ans = 0 for i in range(n): for j in range(n): if dp[i][j]: if i <= j: ans = max(ans, s[j+1]-s[i]) else: ans = max(ans, s[n]-s[i]+s[j+1]) print(ans)