結果

問題 No.2423 Merge Stones
ユーザー t98slider
提出日時 2023-08-14 20:11:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,403 bytes
コンパイル時間 1,988 ms
コンパイル使用メモリ 178,400 KB
実行使用メモリ 317,440 KB
最終ジャッジ日時 2024-11-22 20:37:43
合計ジャッジ時間 164,249 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 54 TLE * 18
権限があれば一括ダウンロードができます

ソースコード

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