結果

問題 No.2423 Merge Stones
ユーザー t98slider
提出日時 2023-08-14 21:43:25
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 728 ms / 4,000 ms
コード長 959 bytes
コンパイル時間 404 ms
コンパイル使用メモリ 82,324 KB
実行使用メモリ 82,464 KB
最終ジャッジ日時 2024-11-22 22:40:27
合計ジャッジ時間 41,262 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 72
権限があれば一括ダウンロードができます

ソースコード

diff # 

from sys import stdin
import itertools
N, K = map(int, stdin.readline().split())
A = list(map(int, stdin.readline().split()))
C = list(map(int, stdin.readline().split()))
dp = [[0] * (2 * N + 1) for _ in range(2 * N)]
dp2 = [[0] * (2 * N + 1) for _ in range(2 * N)]
s = [0] + list(itertools.accumulate(A + A))
ans = max(A)
coef = (1 << (K + 1)) - 1
for i in range(N):
    C[i] -= 1
    dp[i][i + 1] |= 1 << C[i]
    dp2[i][i + 1] = coef * dp[i][i + 1]
    dp2[i][i + 1] |= dp2[i][i + 1] >> K
    dp[i + N][i + 1 + N] = dp[i][i + 1]
    dp2[i + N][i + 1 + N] = dp2[i][i + 1]

for Len in range(2, N + 1):
    for l in range(2 * N - Len):
        r = l + Len
        for m in range(l, r):
            dp[l][r] |= dp[l][m] & dp2[m][r]
            dp[l][r] |= dp2[l][m] & dp[m][r]
        
        if dp[l][r]:
            ans = max(ans, s[r] - s[l])
            for j in range(K + 1): dp2[l][r] |= dp[l][r] << j
            dp2[l][r] |= dp2[l][r] >> K

print(ans)
0