結果

問題 No.2423 Merge Stones
コンテスト
ユーザー t98slider
提出日時 2023-08-14 20:23:35
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,327 bytes
コンパイル時間 1,819 ms
コンパイル使用メモリ 176,376 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-22 21:02:46
合計ジャッジ時間 19,923 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 21 WA * 51
権限があれば一括ダウンロードができます

ソースコード

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);
    vector<ll> sa(n2 + 1);
    cin >> a >> c;
    a.insert(a.end(), a.begin(), a.end());
    c.insert(c.end(), c.begin(), c.end());
    for(auto &&v : c) v--;
    for(int i = 0; i < n2; i++) sa[i + 1] = sa[i] + a[i];
    vector<vector<ll>> dp(n2 + 1, vector<ll>(n2 + 1));
    for(int i = 0; i < n2; i++) dp[i][i + 1] = 1ll << c[i];
    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] & dp[m][r];
                for(int i = 1; i <= k; i++){
                    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];
                }
            }
            if(dp[l][r]) ans = max(ans, sa[r] - sa[l]);
        }
    }
    cout << ans << '\n';
}
0