結果

問題 No.2139 K Consecutive Sushi
ユーザー HIcoderHIcoder
提出日時 2024-10-09 23:15:49
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 152 ms / 2,000 ms
コード長 2,570 bytes
コンパイル時間 1,380 ms
コンパイル使用メモリ 121,100 KB
実行使用メモリ 10,568 KB
最終ジャッジ日時 2024-10-09 23:15:54
合計ジャッジ時間 5,777 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 119 ms
10,568 KB
testcase_04 AC 129 ms
10,496 KB
testcase_05 AC 121 ms
10,480 KB
testcase_06 AC 115 ms
10,552 KB
testcase_07 AC 122 ms
10,468 KB
testcase_08 AC 152 ms
10,428 KB
testcase_09 AC 126 ms
10,428 KB
testcase_10 AC 127 ms
10,440 KB
testcase_11 AC 127 ms
10,432 KB
testcase_12 AC 123 ms
10,496 KB
testcase_13 AC 2 ms
6,820 KB
testcase_14 AC 2 ms
6,816 KB
testcase_15 AC 2 ms
6,816 KB
testcase_16 AC 2 ms
6,820 KB
testcase_17 AC 2 ms
6,816 KB
testcase_18 AC 3 ms
6,820 KB
testcase_19 AC 3 ms
6,820 KB
testcase_20 AC 3 ms
6,816 KB
testcase_21 AC 2 ms
6,820 KB
testcase_22 AC 4 ms
6,820 KB
testcase_23 AC 13 ms
6,820 KB
testcase_24 AC 41 ms
6,816 KB
testcase_25 AC 88 ms
9,328 KB
testcase_26 AC 13 ms
6,816 KB
testcase_27 AC 34 ms
6,816 KB
testcase_28 AC 35 ms
6,816 KB
testcase_29 AC 18 ms
6,820 KB
testcase_30 AC 65 ms
6,816 KB
testcase_31 AC 113 ms
10,076 KB
testcase_32 AC 19 ms
6,816 KB
testcase_33 AC 103 ms
10,460 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<string>
#include<queue>
#include<vector>
#include<cassert>
#include<random>
#include<set>
#include<map>
#include<cassert>
#include<unordered_map>
#include<bitset>
#include<numeric>
#include<algorithm>
using namespace std;
using ll = long long;
const int inf=1<<30;
const ll INF=1LL<<62;
typedef pair<int,ll> P;
typedef pair<int,P> PP; 
const ll MOD=998244353;


ll op(ll x, ll y){
    //(x,y)を比較する任意の演算子.今回はmaxとした
    return min(x,y);
}

ll e(){
    //initialize value
    return INF;
}


template<class Type,
    Type (*op)(Type,Type),
    Type (*e)()
    > class segment_tree{
    
    public:
        
        std::vector<Type> dat;
        int n=1;
        segment_tree(int n_){    
            while(n < n_){
                n*=2;
            }
            dat = std::vector<Type>(2*n-1,e());// 0,1,2,...,2*n-1,2*n-2
        }

        ~segment_tree(){
            std::vector<Type>().swap(dat);
        }

        void set(int k,Type a){
            update(k,a);
        }


        void update(int k,Type a){
            k+=n-1;
            dat[k] = a;
            while(k>0){
                k = (k-1)/2;
                dat[k]=op(dat[2*k+1],dat[2*k+2]);
            }
        }

        //[a,b)
        Type query(int a,int b,int k,int l,int r){
            if(r<=a || b<=l) return e();
            if(a<=l && r<=b) return dat[k];
            else{

                Type vl = query(a,b,2*k+1,l,(l+r)/2);
                Type vr = query(a,b,2*k+2,(l+r)/2,r);
                return op(vl,vr);
            }
        }

        //[a,b)の範囲でのmax
        Type query(int a,int b){
            return query(a,b,0,0,n);
        }

        Type operator[](int index){
            
            return get(index);
        }

        Type get(int index){
            index += n-1;
            return dat[index];
        }

        void add(int index,Type val){
            update(index,op(get(index),val));
        }
};


int main(){
    int N,K;
    cin>>N>>K;
    vector<ll> A(N+2);
    ll totA=0;
    for(int i=1;i<=N;i++){
        cin>>A[i];
        totA+=A[i];
    }
    vector<ll> dp(N+2);
    segment_tree<ll,op,e> seg(N+2);
    dp[0]=0;
    seg.update(0,0);
    for(int i=1;i<=N+1;i++){
        //[i-K,i)
        //A[i]を必ずとる
        dp[i]=seg.query(max(0,i-K),i)+A[i];
        seg.update(i,dp[i]);
    }

    // for(int i=1;i<=N+1;i++){
    //     cout<<"dp["<<i<<"]="<<dp[i]<<endl;
    // }

    // cout<<"totA="<<totA<<endl;

    ll ans=totA-dp[N+1];
    cout<<ans<<endl;



    
}
0