結果
問題 | No.2139 K Consecutive Sushi |
ユーザー |
![]() |
提出日時 | 2022-12-03 00:19:51 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 298 ms / 2,000 ms |
コード長 | 1,780 bytes |
コンパイル時間 | 2,220 ms |
コンパイル使用メモリ | 202,472 KB |
最終ジャッジ日時 | 2025-02-09 04:32:56 |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
#include <bits/stdc++.h> using namespace std; template <typename T> struct LazySegmentTree { int n; vector<T> dat; vector<T>lazy; LazySegmentTree() {}; LazySegmentTree(int N) { n = 1; while(n < N) { n *= 2; } dat.resize(n*2,0); lazy.resize(n*2,0); } void eval(int k,int l,int r) { if(lazy[k] == 0) return; if(k < n) { lazy[k*2] += lazy[k]; lazy[k*2+1] += lazy[k]; } dat[k] += lazy[k]; lazy[k] = 0; } void update(int a, int b, T x, int k, int l, int r) { eval(k,l,r); if(a <= l && r <= b) { lazy[k] += x; eval(k,l,r); } else if(a < r && l < b) { update(a, b, x, k*2, l, (l+r)/2); update(a, b, x, k*2+1, (l+r)/2, r); dat[k] = max(dat[k*2],dat[k*2+1]); } } void update(int a, int b, T x) {//[a,b) update(a, b, x, 1, 0, n); } T query_sub(int a, int b, int k, int l, int r) { eval(k,l,r); if (r <= a || b <= l) { return 0; } else if (a <= l && r <= b) { return dat[k]; } else { T vl = query_sub(a, b, k*2, l, (l+r)/2); T vr = query_sub(a, b, k*2+1, (l+r)/2, r); return max(vl,vr); } } T query(int a, int b) {//[a,b) return query_sub(a, b, 1, 0, n); } }; int main() { int N,K; cin >> N >> K; vector<int>A(N); LazySegmentTree<long long>Seg(N+1); for(int i = 0; i < N; i++) { cin >> A[i]; Seg.update(i+1,i+2,Seg.query(max(i-K+1,0),i+1)); Seg.update(max(i-K+2,0),i+1,A[i]); } cout << Seg.query(max(0,N-K+1),N+1) << endl; }