結果

問題 No.1117 数列分割
コンテスト
ユーザー fumofumofuni
提出日時 2021-12-30 13:22:07
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 600 ms / 3,000 ms
コード長 2,694 bytes
コンパイル時間 2,290 ms
コンパイル使用メモリ 207,536 KB
最終ジャッジ日時 2025-01-27 07:37:59
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
using namespace std;
#define rep(i,n) for(ll i=0;i<n;i++)
#define repl(i,l,r) for(ll i=(l);i<(r);i++)
#define per(i,n) for(ll i=(n)-1;i>=0;i--)
#define perl(i,r,l) for(ll i=r-1;i>=l;i--)
#define fi first
#define se second
#define pb push_back
#define ins insert
#define pqueue(x) priority_queue<x,vector<x>,greater<x>>
#define all(x) (x).begin(),(x).end()
#define CST(x) cout<<fixed<<setprecision(x)
#define vtpl(x,y,z) vector<tuple<x,y,z>>
#define rev(x) reverse(x);
using ll=long long;
using vl=vector<ll>;
using vvl=vector<vector<ll>>;
using pl=pair<ll,ll>;
using vpl=vector<pl>;
using vvpl=vector<vpl>;
const ll MOD=1000000007;
const ll MOD9=998244353;
const int inf=1e9+10;
const ll INF=4e17;
const ll dy[9]={0,1,-1,0,1,1,-1,-1,0};
const ll dx[9]={1,0,0,-1,1,-1,1,-1,0};
template<class T> inline bool chmin(T& a, T b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}
template<class T> inline bool chmax(T& a, T b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
}

template<class T>
class foldable_queue{
    function<T(T,T)> fx;
    using P=pair<T,T>;
    stack<P> sf,sb;
    T ex;
    void reload(){
        while(!sb.empty()){
            if(sf.empty())sf.push(make_pair(sb.top().first,sb.top().first));
            else sf.push(make_pair(sb.top().first,fx(sb.top().first,sf.top().second)));
            sb.pop();
        }
    }
    public:
    foldable_queue(function<T(T,T)> _fx,T _ex){
        fx=_fx;ex=_ex;
    }
    int size(){
        return sf.size()+sb.size();
    }
    T front(){
        if(sf.empty())reload();
        return sf.top().first;
    }
    T fold(){
        T ans=(sf.empty()? ex:sf.top().second);
        ans=fx(ans,(sb.empty()? ex:sb.top().second));
        return ans;
    }
    void push(T x){
        if(sb.empty())sb.push(make_pair(x,x));
        else sb.push(make_pair(x,fx(sb.top().second,x)));
    }
    void pop(){
        if(sf.empty())reload();
        sf.pop();
    }
};


int main(){
    ll n,k,m;cin >> n >> k >> m;
    vl a(n);rep(i,n)cin >> a[i];
    vl cum(n+1);rep(i,n)cum[i+1]+=cum[i]+a[i];
    vl dp(n+1,-INF);dp[0]=0;
    rep(_,k){//すでに_個決めた。
        vl ndp(n+1,-INF);
        foldable_queue<ll> que([](ll a,ll b){return max(a,b);},-INF);
        rep(i,n+1){
            if(que.size()>m)que.pop();
            chmax(ndp[i],que.fold()+cum[i]);
            que.push(dp[i]-cum[i]);
        }
        while(que.size())que.pop();
        rep(i,n+1){
            if(que.size()>m)que.pop();
            chmax(ndp[i],que.fold()-cum[i]);
            que.push(dp[i]+cum[i]);
        }
        swap(dp,ndp);
    }
    cout << dp.back() << endl;
}
0