結果

問題 No.2423 Merge Stones
ユーザー firiexp
提出日時 2023-07-15 13:31:48
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 634 ms / 4,000 ms
コード長 1,663 bytes
コンパイル時間 2,846 ms
コンパイル使用メモリ 125,964 KB
最終ジャッジ日時 2025-02-15 14:59:01
ジャッジサーバーID
(参考情報)
judge4 / judge5
純コード判定しない問題か言語
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 72
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <map>
#include <queue>

static const int MOD = 1000000007;
using ll = long long;
using u32 = unsigned;
using namespace std;

template<class T> constexpr T INF = ::numeric_limits<T>::max() / 32 * 15 + 208;

ll dp[600][601];
int main() {
    int n, k;
    cin >> n >> k;
    vector<int> A(2*n), C(2*n);
    for (int i = 0; i < n; ++i) {
        scanf("%d", &A[i]);
        A[i+n] = A[i];
    }
    vector<ll> S(2*n+1);
    for (int i = 0; i < 2*n; ++i) {
        S[i+1] = S[i] + A[i];
    }
    for (int i = 0; i < n; ++i) {
        scanf("%d", &C[i]);
        C[i+n] = C[i];
    }
    n *= 2;
    for (int i = 0; i < n; ++i) {
        for (int j = i+1; j <= n; ++j) {
            dp[i][j] = 0;
        }
        dp[i][i+1] = (1LL << C[i]);
    }

    for (int len = 2; len <= n; ++len) {
        for (int l = 0; l+len <= n; ++l) {

            int r = l+len;
            for (int mid = l+1; mid < r; ++mid) {
                ll lval = dp[l][mid], rval = dp[mid][r];
                if(lval == 0 || rval == 0) continue;
                dp[l][r] |= lval & rval;
                for (int i = 1; i <= k; ++i) {
                    dp[l][r] |= lval & (rval << i);
                    dp[l][r] |= lval & (rval >> i);
                    dp[l][r] |= lval << i & rval;
                    dp[l][r] |= lval >> i & rval;
                }
            }
        }
    }
    ll ans = 0;
    for (int i = 0; i < n; ++i) {
        for (int j = i+1; j <= n && j-i <= n/2; ++j) {
            if(dp[i][j] > 0) ans = max(ans, S[j]-S[i]);
        }
    }
    cout << ans << "\n";
    return 0;
}
0