結果

問題 No.1117 数列分割
ユーザー mugen_1337
提出日時 2020-07-18 17:24:26
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,800 bytes
コンパイル時間 2,238 ms
コンパイル使用メモリ 201,676 KB
最終ジャッジ日時 2025-01-12 00:19:39
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 13 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define ALL(x) x.begin(),x.end()
#define rep(i,n) for(int i=0;i<(n);i++)
#define debug(v) cout<<#v<<":";for(auto x:v){cout<<x<<' ';}cout<<endl;
#define mod 1000000007
using ll=long long;
const int INF=1000000000;
const ll LINF=1001002003004005006ll;
int dx[]={1,0,-1,0};
int dy[]={0,1,0,-1};
// ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(b<a){a=b;return true;}return false;}

struct IOSetup{
    IOSetup(){
        cin.tie(0);
        ios::sync_with_stdio(0);
        cout<<fixed<<setprecision(12);
    }
} iosetup;

template<typename T1,typename T2>
ostream &operator<<(ostream &os,const pair<T1,T2>&p){
    os<<p.first<<" "<<p.second;
    return os;
}

template<typename T>
ostream &operator<<(ostream &os,const vector<T>&v){
    for(int i=0;i<(int)v.size();i++) os<<v[i]<<(i+1==v.size()?"":" ");
    return os;
}

template<typename T>
vector<T> slide_max(const vector<T> &v,int k){
    //idxとvalueが昇順になるようにdequeに入れる
    deque<int> dq;
    vector<T> ret;
    for(int i=0;i<v.size();i++){
        //不等号: <=ならslide max  >=ならslide minになる
        while(!dq.empty()&&v[dq.back()]>=v[i]){
            dq.pop_back();
        }
        dq.push_back(i);
        {
            ret.push_back(v[dq.front()]);
            if(dq.front()==i-k+1) dq.pop_front();
        }
    }
    return ret;
}

ll dp[3010][3010];
ll a[3010],sum[3010];

signed main(){
    rep(i,3010)rep(j,3010)dp[i][j]=-LINF;
    dp[0][0]=0;

    int n,k,m;cin>>n>>k>>m;
    rep(i,n){
        cin>>a[i];
        sum[i+1]=a[i]+sum[i];
    }

    rep(j,k){
        deque<ll> d1,d2;// dp-sum,dp+sum
        vector<ll>sm1,sm2;
        rep(i,n+1){
            while(!d1.empty() and dp[d1.back()][j]-sum[d1.back()]<=dp[i][j]-sum[i]) d1.pop_front();
            d1.push_back(i);
            sm1.push_back(dp[d1.front()][j]-sum[d1.front()]);
            if(d1.front()==i-m+1) d1.pop_front();
        }
        rep(i,n+1){
            while(!d2.empty() and dp[d2.back()][j]+sum[d2.back()]<=dp[i][j]+sum[i]) d2.pop_front();
            d2.push_back(i);
            sm2.push_back(dp[d2.front()][j]+sum[d2.front()]);
            if(d2.front()==i-m+1) d2.pop_front();
        }
        // cout<<j<<endl;
        // rep(i,n+1) cout<<dp[i][j]<<" ";cout<<endl;
        // debug(sm1);
        // debug(sm2);

        rep(i,n){
            chmax(dp[i+1][j+1],sm1[i]+sum[i+1]);
            chmax(dp[i+1][j+1],sm2[i]-sum[i+1]);
        }
    }

    // rep(j,k+1){
    //     cout<<j<<" : ";
    //     rep(i,n+1){
    //         cout<<dp[i][j]<<" ";
    //     }
    //     cout<<endl;
    // }

    cout<<dp[n][k]<<endl;
    return 0;
}
0