結果

問題 No.2423 Merge Stones
コンテスト
ユーザー titia
提出日時 2023-08-24 01:04:40
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 618 ms / 4,000 ms
コード長 838 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,571 ms
コンパイル使用メモリ 85,120 KB
実行使用メモリ 81,792 KB
最終ジャッジ日時 2026-05-28 19:20:42
合計ジャッジ時間 37,789 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge1_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 72
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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