結果

問題 No.2423 Merge Stones
コンテスト
ユーザー lam6er
提出日時 2025-04-15 22:04:42
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,088 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 201 ms
コンパイル使用メモリ 81,696 KB
実行使用メモリ 87,372 KB
最終ジャッジ日時 2025-04-15 22:06:36
合計ジャッジ時間 6,448 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 10 TLE * 1 -- * 61
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def main():
    import sys
    input = sys.stdin.read().split()
    idx = 0
    N = int(input[idx])
    idx += 1
    K = int(input[idx])
    idx += 1
    A = list(map(int, input[idx:idx+N]))
    idx += N
    C = list(map(int, input[idx:idx+N]))
    
    # Duplicate the array to handle circularity
    A += A
    C += C
    
    # Precompute allowed masks for each color (1-based)
    max_color = 50
    allowed_masks = [0] * (max_color + 2)  # 0..50
    for c in range(1, max_color + 1):
        lower = max(1, c - K)
        upper = min(max_color, c + K)
        mask = 0
        for d in range(lower, upper + 1):
            mask |= 1 << d
        allowed_masks[c] = mask
    
    # Precompute prefix sums
    prefix = [0] * (2 * N + 1)
    for i in range(2 * N):
        prefix[i + 1] = prefix[i] + A[i]
    
    # Initialize DP table
    dp = [[0] * (2 * N) for _ in range(2 * N)]
    for i in range(2 * N):
        c = C[i]
        dp[i][i] = 1 << c
    
    # Fill DP table
    for l in range(2, N + 1):
        for i in range(2 * N - l + 1):
            j = i + l - 1
            current_mask = 0
            for k in range(i, j):
                left = dp[i][k]
                right = dp[k + 1][j]
                if left == 0 or right == 0:
                    continue
                # Iterate over all possible colors in left
                c1 = 1
                while c1 <= max_color:
                    if (left & (1 << c1)) == 0:
                        c1 += 1
                        continue
                    allowed = allowed_masks[c1]
                    overlap = right & allowed
                    if overlap != 0:
                        current_mask |= (1 << c1) | overlap
                    c1 += 1
            dp[i][j] = current_mask
    
    # Find the maximum sum
    max_sum = 0
    for i in range(2 * N):
        for j in range(i, min(i + N, 2 * N)):
            if dp[i][j] != 0:
                s = prefix[j + 1] - prefix[i]
                if s > max_sum:
                    max_sum = s
    print(max_sum)

if __name__ == '__main__':
    main()
0