結果

問題 No.2139 K Consecutive Sushi
コンテスト
ユーザー otoshigo
提出日時 2022-12-03 03:55:13
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 73 ms / 2,000 ms
コード長 1,773 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,075 ms
コンパイル使用メモリ 199,240 KB
最終ジャッジ日時 2025-02-09 04:39:16
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
#define rep(i,n) for(int i=0;i<n;i++)
#define rrep(i,n) for(int i=(n)-1;i>=0;i--)
#define all(v) v.begin(),v.end()
#define rall(v) v.rbegin(),v.rend()
template<class T> bool chmax(T &a, T b){if (a < b){a = b;return true;} else return false;}
template<class T> bool chmin(T &a, T b){if (a > b){a = b;return true;} else return false;}

template <class S, S (*op)(S, S), S (*e)()>
struct segtree{
    int n;
    vector<S> dat;
    segtree(int N){
        int x = 1;
        while (x < N) x *= 2;
        n = x;
        dat.resize(2 * n - 1, e());
    }
    void set(int i, S x){
        assert(i >= 0 && i < n);
        i += n - 1;
        dat[i] = x;
        while (i > 0){
            i = (i - 1) / 2;
            dat[i] = op(dat[2 * i + 1], dat[2 * i + 2]);
        }
    }
    S get(int i){
        assert(i >= 0 && i < n);
        i += n - 1;
        return dat[i];
    }
    S prod(int a, int b, int k = 0, int l = 0, int r = -1){
        if (r < 0) r = n;
        if (r <= a || l >= b) return e();
        if (l >= a && r <= b) return dat[k];
        else{
            S vl = prod(a, b, 2 * k + 1, l, (l + r) / 2);
            S vr = prod(a, b, 2 * k + 2, (l + r) / 2, r);
            return op(vl, vr);
        }
    }
};

ll op(ll l,ll r){
    return min(l,r);
}

ll e(){
    return 1LL<<60;
}

int n,k;
ll A[200010];
int main(){
    ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    cin>>n>>k;
    rep(i,n)cin>>A[i];
    segtree<ll,op,e>seg(n);
    rep(i,k)seg.set(i,A[i]);
    for(int i=k;i<n;i++){
        ll m=seg.prod(i-k,i);
        seg.set(i,m+A[i]);
    }
    ll ans=1LL<<60;
    for(int i=n-k;i<n;i++)chmin(ans,seg.get(i));
    ans=-ans;
    rep(i,n)ans+=A[i];
    cout<<ans<<"\n";
}
0