結果

問題 No.2423 Merge Stones
ユーザー titiatitia
提出日時 2023-08-24 01:04:40
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,174 ms / 4,000 ms
コード長 838 bytes
コンパイル時間 251 ms
コンパイル使用メモリ 82,852 KB
実行使用メモリ 77,056 KB
最終ジャッジ日時 2024-12-22 13:13:39
合計ジャッジ時間 62,114 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 72
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

N,K=map(int,input().split())
A=list(map(int,input().split()))
C=list(map(int,input().split()))

DP=[[0]*N for i in range(N)]

for i in range(N):
    DP[i][0]=1<<C[i]

for j in range(1,N):
    for i in range(N):
        # DP[i][j]
        # [i,i+j]

        for k in range(j):
            # [i,i+k], [i+k+1,j]
            # DP[i][k], DP[i+k+1][j-k]

            a=DP[i][k]
            b=DP[(i+k+1)%N][j-k-1]

            ANS=0

            for l in range(K+1):
                ANS|=(a&(b<<l))
                ANS|=(a&(b>>l))
                ANS|=(b&(a<<l))
                ANS|=(b&(a>>l))

            DP[i][j]|=ANS
            
    
S=[0]
for a in A+A:
    S.append(S[-1]+a)

ANS=0
for i in range(N):
    for j in range(N):
        if DP[i][j]!=0:
            ANS=max(ANS,S[i+j+1]-S[i])

print(ANS)
0