結果

問題 No.2423 Merge Stones
コンテスト
ユーザー t98slider
提出日時 2023-08-14 21:38:58
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 61 ms / 4,000 ms
コード長 1,278 bytes
コンパイル時間 1,717 ms
コンパイル使用メモリ 173,904 KB
実行使用メモリ 9,268 KB
最終ジャッジ日時 2024-11-22 22:33:50
合計ジャッジ時間 7,030 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 72
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

template<class T> istream& operator >> (istream& is, vector<T>& vec) {
    for(T& x : vec) is >> x;
    return is;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n, k, n2;
    cin >> n >> k;
    n2 = 2 * n;
    vector<int> a(n), c(n);
    ll sa[n2 + 1] = {}, dp[n2][n2 + 1] = {}, dp2[n2][n2 + 1] = {};
    cin >> a >> c;
    a.insert(a.end(), a.begin(), a.end());
    c.insert(c.end(), c.begin(), c.end());
    for(int i = 0; i < n2; i++){
        sa[i + 1] = sa[i] + a[i];
        dp[i][i + 1] = 1ll << --c[i];
        dp2[i][i + 1] = ((1 << k + 1) - 1) * dp[i][i + 1];
        dp2[i][i + 1] |= dp2[i][i + 1] >> k;
    }
    ll ans = *max_element(a.begin(), a.end());
    for(int len = 2; len <= n; len++){
        for(int l = 0; l + len < n2; l++){
            int r = l + len;
            for(int m = l + 1; m < r; m++){
                dp[l][r] |= dp[l][m] & dp2[m][r];
                dp[l][r] |= dp2[l][m] & dp[m][r];
            }
            if(dp[l][r]){
                ans = max(ans, sa[r] - sa[l]);
                for(int i = 0; i <= k; i++) dp2[l][r] |= dp[l][r] << i;
                dp2[l][r] |= dp2[l][r] >> k;
            }
        }
    }
    cout << ans << '\n';
}
0