結果

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

ソースコード

diff # 

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

int main() {
    // 入力 (数珠なので、2 週させる)
    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
    vector dp(N*2, vector(N*2+1, bitset<52>(false)));
    for (int l = 0; l < N*2; ++l) dp[l][l+1][C[l]] = true;
    for (int between = 2; between <= N; ++between) {
        for (int l = 0; l + between <= N*2; ++l) {
            int r = l + between;
            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];
                }
            }
        }
    }
    
    // 集計
    long long res = 0;
    for (int l = 0; l <= N; ++l) {
        for (int r = l+1; r <= N*2; ++r) {
            for (int c = 0; c <= 50; ++c) {
                if (dp[l][r][c]) res = max(res, S[r] - S[l]);
            }
        }
    }
    cout << res << endl;
}
0