結果

問題 No.2423 Merge Stones
ユーザー drken1215
提出日時 2023-08-12 19:32:25
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,149 bytes
コンパイル時間 2,161 ms
コンパイル使用メモリ 201,892 KB
最終ジャッジ日時 2025-02-16 07:32:13
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 70 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

int main() {
    // 入力
    int N, K;
    cin >> N >> K;
    vector<long long> A(N*2), C(N*2);
    for (int i = 0; i < N; ++i) { cin >> A[i]; A[i+N] = A[i]; }
    for (int i = 0; i < N; ++i) { cin >> C[i]; C[i+N] = C[i]; }
    
    // 累積和
    vector<long long> S(N*2+1, 0);
    for (int i = 0; i < N*2; ++i) S[i+1] = S[i] + A[i];
    
    // dp
    long long res = 0;
    vector dp(N*2+1, vector(N*2+2, bitset<52>(false)));
    for (int l = 0; l < N*2; ++l) dp[l][l+1][C[l]] = true;
    for (int b = 2; b <= N; ++b) {
        for (int l = 0; l < N*2 && l+b <= N*2; ++l) {
            int r = l + b;
            for (int m = l+1; m < r; ++m) {
                for (int k = 0; k <= K; ++k) {
                    dp[l][r] |= dp[l][m] & (dp[m][r] << k);
                    dp[l][r] |= dp[l][m] & (dp[m][r] >> k);
                    dp[l][r] |= (dp[l][m] << k) & dp[m][r];
                    dp[l][r] |= (dp[l][m] >> k) & dp[m][r];
                }
            }
            for (int c = 0; c <= 50; ++c) if (dp[l][r][c]) res = max(res, S[r] - S[l]);
        }
    }
    cout << res << endl;
}
0