結果

問題 No.1117 数列分割
ユーザー mugen_1337mugen_1337
提出日時 2020-07-21 13:17:23
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 782 ms / 3,000 ms
コード長 2,665 bytes
コンパイル時間 4,464 ms
コンパイル使用メモリ 207,780 KB
実行使用メモリ 74,596 KB
最終ジャッジ日時 2023-08-28 09:12:24
合計ジャッジ時間 11,775 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 22 ms
74,172 KB
testcase_01 AC 22 ms
74,228 KB
testcase_02 AC 21 ms
74,200 KB
testcase_03 AC 71 ms
74,196 KB
testcase_04 AC 64 ms
74,164 KB
testcase_05 AC 22 ms
74,456 KB
testcase_06 AC 23 ms
74,168 KB
testcase_07 AC 22 ms
74,224 KB
testcase_08 AC 56 ms
74,532 KB
testcase_09 AC 32 ms
74,284 KB
testcase_10 AC 58 ms
74,204 KB
testcase_11 AC 177 ms
74,268 KB
testcase_12 AC 236 ms
74,264 KB
testcase_13 AC 256 ms
74,256 KB
testcase_14 AC 266 ms
74,308 KB
testcase_15 AC 413 ms
74,272 KB
testcase_16 AC 445 ms
74,424 KB
testcase_17 AC 55 ms
74,356 KB
testcase_18 AC 649 ms
74,252 KB
testcase_19 AC 782 ms
74,208 KB
testcase_20 AC 353 ms
74,252 KB
testcase_21 AC 441 ms
74,328 KB
testcase_22 AC 618 ms
74,356 KB
testcase_23 AC 621 ms
74,596 KB
testcase_24 AC 609 ms
74,316 KB
testcase_25 AC 607 ms
74,408 KB
testcase_26 AC 22 ms
74,188 KB
testcase_27 AC 46 ms
74,216 KB
testcase_28 AC 47 ms
74,196 KB
権限があれば一括ダウンロードができます

ソースコード

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;
}

// Semigroup Sliding-Window-AGregation
template<typename Semigroup>
struct SWAG{
    using F=function<Semigroup(Semigroup,Semigroup)>;
    const F f;
    stack<pair<Semigroup,Semigroup>> fr,ba;
    SWAG(const F f):f(f){}
    Semigroup get(){
        if(fr.empty()) return ba.top().second;
        if(ba.empty()) return fr.top().second;
        return f(fr.top().second,ba.top().second);
    }
    void push(Semigroup x){
        if(ba.empty()) ba.push(make_pair(x,x));
        else ba.push(make_pair(x,f(ba.top().second,x)));
    }
    void pop(){
        if(empty()) return ;
        if(!fr.empty()) fr.pop();
        else{
            while(!ba.empty()){
                pair<Semigroup,Semigroup> p=ba.top();ba.pop();
                if(fr.empty()) fr.push(make_pair(p.first,p.first));
                else fr.push(make_pair(p.first,f(p.first,fr.top().second)));
            }
            fr.pop();
        }
    }
    size_t size(){
        return fr.size()+ba.size();
    }
    bool empty(){
        return size()==0;
    }
};

ll dp[3010][3010],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]=sum[i]+a[i];
    }

    auto f=[](ll a,ll b){return a>b?a:b;};
    rep(j,k){
        SWAG<ll> s1(f),s2(f);
        rep(i,n){
            if(s1.size()>=m) s1.pop();
            if(s2.size()>=m) s2.pop();
            s1.push(dp[i][j]-sum[i]);
            s2.push(dp[i][j]+sum[i]);

            chmax(dp[i+1][j+1],s1.get()+sum[i+1]);
            chmax(dp[i+1][j+1],s2.get()-sum[i+1]);
        }
    }
    
    cout<<dp[n][k]<<endl;
    return 0;
}
0