結果

問題 No.2139 K Consecutive Sushi
ユーザー umezoumezo
提出日時 2022-12-03 00:55:07
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 266 ms / 2,000 ms
コード長 2,192 bytes
コンパイル時間 2,400 ms
コンパイル使用メモリ 205,640 KB
実行使用メモリ 18,944 KB
最終ジャッジ日時 2024-04-18 10:01:56
合計ジャッジ時間 7,133 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 228 ms
18,944 KB
testcase_04 AC 266 ms
18,944 KB
testcase_05 AC 241 ms
18,816 KB
testcase_06 AC 213 ms
18,816 KB
testcase_07 AC 242 ms
18,816 KB
testcase_08 AC 248 ms
18,816 KB
testcase_09 AC 250 ms
18,696 KB
testcase_10 AC 262 ms
18,912 KB
testcase_11 AC 255 ms
18,816 KB
testcase_12 AC 239 ms
18,808 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 1 ms
5,376 KB
testcase_18 AC 3 ms
5,376 KB
testcase_19 AC 3 ms
5,376 KB
testcase_20 AC 3 ms
5,376 KB
testcase_21 AC 1 ms
5,376 KB
testcase_22 AC 4 ms
5,376 KB
testcase_23 AC 22 ms
5,376 KB
testcase_24 AC 78 ms
8,192 KB
testcase_25 AC 184 ms
13,556 KB
testcase_26 AC 20 ms
5,376 KB
testcase_27 AC 65 ms
7,424 KB
testcase_28 AC 61 ms
7,680 KB
testcase_29 AC 29 ms
5,376 KB
testcase_30 AC 85 ms
8,704 KB
testcase_31 AC 236 ms
17,152 KB
testcase_32 AC 30 ms
5,632 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;

#include <bits/stdc++.h>
using namespace std;

template <typename X,typename M>
struct SegTreeLazy {
  using FX=function<X(X,X)>;
  using FA=function<X(X,M)>;
  using FM=function<M(M,M)>;
  int n;
  FX fx;
  FA fa;
  FM fm;
  const X ex;
  const M em;
  vector<X> dat;
  vector<M> lazy;
  SegTreeLazy(int n_,FX fx_,FA fa_,FM fm_,X ex_,M em_)
    : n(),fx(fx_),fa(fa_),fm(fm_),ex(ex_),em(em_),dat(n_*4,ex),lazy(n_*4,em) {
    int x=1;
    while (n_ > x) x *= 2;
    n=x;
  }
  void set(int i,X x) { dat[i+n-1]=x;}
  void build() {
    for (int k=n-2;k >= 0;k--) dat[k]=fx(dat[2*k+1],dat[2*k+2]);
  }

  void eval(int k) {
    if(lazy[k] == em) return;
    if(k<n-1) {           
      lazy[k*2+1]=fm(lazy[k*2+1],lazy[k]);
      lazy[k*2+2]=fm(lazy[k*2+2],lazy[k]);
    }
    
    dat[k]=fa(dat[k],lazy[k]);
    lazy[k]=em;
  }
  void update(int a,int b,M x,int k,int l,int r){
    eval(k);
    if(a<=l && r<=b){ 
      lazy[k]=fm(lazy[k],x);
      eval(k);
    }else if(a<r && l<b){                    
      update(a,b,x,k*2+1,l,(l+r)/2); 
      update(a,b,x,k*2+2,(l+r)/2,r); 
      dat[k]=fx(dat[k*2+1],dat[k*2+2]);
    }
  }
  void update(int a,int b,M x){update(a,b,x,0,0,n);}
  X query_sub(int a,int b,int k,int l,int r){
    eval(k);
    if(r<=a || b<=l){ 
      return ex;
    }else if(a <= l && r <= b){ 
      return dat[k];
    }else{ 
      X vl=query_sub(a,b,k*2+1,l,(l+r)/2);
      X vr=query_sub(a,b,k*2+2,(l+r)/2,r);
      return fx(vl,vr);
    }
  }
  X query(int a,int b){return query_sub(a,b,0,0,n);}
};

ll dp[200200];

int main(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);

  int n,k;
  cin>>n>>k;
  vector<ll> A(n);
  rep(i,n) cin>>A[i];
    
  using X=ll;
  using M=ll;
  auto fx=[](X x1,X x2) -> X { return max(x1,x2);};
  auto fa=[](X x,M m) -> X { return x+m;};
  auto fm=[](M m1,M m2) -> M { return m1+m2;};
  X ex=0;
  M em=0;
  SegTreeLazy<X,M> lseg(n+2,fx,fa,fm,ex,em);
  
  for(int i=1;i<=n;i++){
    dp[i]=lseg.query(max(0,i-k),i);
    lseg.update(i,i+1,dp[i]);
    lseg.update(0,i,A[i-1]);
  }
  cout<<lseg.query(n-k+1,n+1)<<endl;

  return 0;
}
0