結果

問題 No.1117 数列分割
コンテスト
ユーザー Shibuyap
提出日時 2020-07-20 22:25:47
言語 C++17
(gcc 15.2.0 + boost 1.89.0)
結果
WA  
実行時間 -
コード長 1,785 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,196 ms
コンパイル使用メモリ 199,576 KB
最終ジャッジ日時 2025-01-12 01:18:14
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 5 WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define srep(i,s,t) for (int i = s; i < t; ++i)
#define drep(i,n) for(int i = (n)-1; i >= 0; --i)
using namespace std;
typedef long long int ll;
typedef pair<int, ll> P;
#define yn {puts("YES");}else{puts("NO");}
#define MAX_N 200005

int main() {
    ll n, k, m;
    cin >> n >> k >> m;
    ll a[n];
    rep(i,n)cin >> a[i];

    ll sum[n+1] = {};
    drep(i,n) sum[i] = sum[i+1] + a[i];

    ll dp[k+1][n+1];
    ll INF = 1001001001001001001;
    rep(i,k+1){
        rep(j,n+1){
            dp[i][j] = -INF;
        }
    }
    
    dp[0][0] = 0;
    

    srep(i,1,k+1){
        deque<P> plus, minus;
        srep(j,1,n+1){
            // 前を削除(pop_front)
            if(!plus.empty() && plus[0].first < j - m){
                plus.pop_front();
            }
            if(!minus.empty() && minus[0].first < j - m){
                minus.pop_front();
            }

            if(i == 1){
                if(j <= m) dp[i][j] = abs(sum[0] - sum[j]);
            }else{
                dp[i][j] = max(dp[i][j], plus[0].second - sum[j]);
                dp[i][j] = max(dp[i][j], minus[0].second + sum[j]);
            }
            
            // 後ろから削除(pop_back)
            ll tmp1 = dp[i-1][j] + sum[j];
            while(!plus.empty() && plus[plus.size()-1].second < tmp1){
                plus.pop_back();
            }
            // 挿入(push_back)
            plus.push_back(P(j,tmp1));

            ll tmp2 = dp[i-1][j] - sum[j];
            while(!minus.empty() && minus[minus.size()-1].second < tmp2){
                minus.pop_back();
            }
            minus.push_back(P(j,tmp2));

        }
    }


    ll ans = dp[k][n];

    cout << ans << endl;
    return 0;
}
0