結果

問題 No.1247 ブロック登り
ユーザー momoyuu
提出日時 2023-09-24 14:31:42
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,036 bytes
コンパイル時間 3,040 ms
コンパイル使用メモリ 251,724 KB
実行使用メモリ 458,044 KB
最終ジャッジ日時 2024-07-17 15:12:44
合計ジャッジ時間 16,952 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 15 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff #

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


ll dp[2][310][310][310];
int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    ll n,k;
    cin>>n>>k;
    vector<ll> a(n);
    for(int i = 0;i<n;i++) cin>>a[i];
    vector<ll> ans(n,-1e18);
    for(int i = 0;i<2;i++) for(int j = 0;j<n;j++) for(int l = 0;l<n;l++) for(int p = 0;p<=k;p++) dp[i][j][l][p] = -1e18;
    for(int i = 0;i<n;i++){
        dp[0][i][i][k-1] = a[i] * k;
        dp[1][i][i][k-1] = a[i] * k;
    }
    for(int i = k;i>=0;i--){
        for(int j = 0;j<n;j++){
            for(int l = j;l<n;l++){
                if(j+1<=l){
                    if(i-2>0){
                        dp[0][j][l][i-2] = max(dp[0][j][l][i-2],dp[0][j][l][i]);
                        dp[1][j][l][i-2] = max(dp[1][j][l][i-2],dp[1][j][l][i]);
                    }
                    if(i%2==1){
                        ans[l] = max(ans[l],dp[0][j][l][i]);
                        ans[j] = max(ans[j],dp[1][j][l][i]);
                    }
                }
                if(i==0){
                    if(l+1<n) ans[l+1] = max(ans[l+1],dp[0][j][l][i]);
                    if(j-1>=0) ans[j-1] = max(ans[j-1],dp[1][j][l][i]);
                }
                if(l+1<n){
                    dp[0][j][l+1][i-1] = max(dp[0][j][l+1][i-1],dp[0][j][l][i]+i*a[l+1]);
                }
                if(j-1>=0){
                    dp[1][j-1][l][i-1] = max(dp[1][j-1][l][i-1],dp[1][j][l][i]+i*a[j-1]);
                }
                int need = l - j;
                if(i-need>=0){
                    dp[0][j][l][i-need] = max(dp[0][j][l][i-need],dp[1][j][l][i]);
                    dp[1][j][l][i-need] = max(dp[1][j][l][i-need],dp[0][j][l][i]);
                }
                int res = i + 1;
                if(j+res<n&&j+res<=l+1) ans[j+res] = max(ans[j+res],dp[1][j][l][i]);
                if(l-res>=0&&l-res>=j-1) ans[l-res] = max(ans[l-res],dp[0][j][l][i]);
            }
        }
    }
    for(int i = 0;i<n;i++) cout<<ans[i]<<endl;
    
}

0