結果

問題 No.2139 K Consecutive Sushi
ユーザー LanatusLanatus
提出日時 2023-03-27 01:49:37
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 127 ms / 2,000 ms
コード長 1,937 bytes
コンパイル時間 1,082 ms
コンパイル使用メモリ 101,024 KB
実行使用メモリ 8,960 KB
最終ジャッジ日時 2024-05-05 19:24:26
合計ジャッジ時間 4,007 ms
ジャッジサーバーID
（参考情報） 		judge2 / 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 118 ms
8,832 KB
testcase_04 AC 127 ms
8,832 KB
testcase_05 AC 121 ms
8,832 KB
testcase_06 AC 113 ms
8,960 KB
testcase_07 AC 119 ms
8,832 KB
testcase_08 AC 123 ms
8,832 KB
testcase_09 AC 124 ms
8,832 KB
testcase_10 AC 126 ms
8,832 KB
testcase_11 AC 125 ms
8,832 KB
testcase_12 AC 122 ms
8,832 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 1 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 3 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 3 ms
5,376 KB
testcase_23 AC 12 ms
5,376 KB
testcase_24 AC 40 ms
5,376 KB
testcase_25 AC 88 ms
8,320 KB
testcase_26 AC 13 ms
5,376 KB
testcase_27 AC 34 ms
5,376 KB
testcase_28 AC 34 ms
5,376 KB
testcase_29 AC 17 ms
5,376 KB
testcase_30 AC 44 ms
5,760 KB
testcase_31 AC 111 ms
8,576 KB
testcase_32 AC 18 ms
5,376 KB
testcase_33 AC 103 ms
8,832 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bitset>
#include <iostream>
#include <vector>
#include <math.h>
#include <set>
#include <map>

using namespace std;

struct UnionFind {
  int n;
  vector<int> parent, siz;
 
  UnionFind(int n) : n(n) {
    parent.resize(n);
    siz.resize(n);
 
    for (int i = 0; i < n; ++i) {
      parent[i] = i;
      siz[i] = 1;
    }
  }
 
  int root(int x) {
    if (parent[x] == x) return x;
    return parent[x] = root(parent[x]);
  }
 
  bool unite(int x, int y) {
    x = root(x);
    y = root(y);
 
    if (x == y) return false;
 
    if (siz[x] < siz[y]) swap(x, y);
    parent[y] = x;
    siz[x] += siz[y];
 
    return true;
  }
 
  int same(int x, int y) {
    return root(x) == root(y);
  }
 
  int size(int x) {
    return siz[root(x)];
  }
};

using ll = long long;

struct RminQ {
  const ll INFLL = 1e18;
  int m;
  vector<long long> dat;
 
  RminQ(int n) {
    m = 1;
    while (m < n) m *= 2;
 
    dat.assign(2 * m - 1, INFLL);
  }
 
  void update(int i, long long x) {
    i += m - 1;
    dat[i] = x;
 
    while (i) {
      i = (i - 1) / 2;
      dat[i] = min(dat[i * 2 + 1], dat[i * 2 + 2]);
    }
  }
 
  // [a, b)
  long long query(int a, int b, int k, int l, int r) {
    if (l >= b || r <= a) return INFLL;
    if (a <= l && r <= b) return dat[k];
    
    long long vl = query(a, b, k * 2 + 1, l, (l + r) / 2);
    long long vr = query(a, b, k * 2 + 2, (l + r) / 2, r);
 
    return min(vl, vr);
  }
 
  long long query_sub(int a, int b) {
    return query(a, b, 0, 0, m);
  }
};

int main(void) {
  int n, k;
  cin >> n >> k;

  vector<long long> a(n + 2, 0);
  for (int i = 1; i <= n; ++i) cin >> a[i];

  RminQ seg(n + 5);
  seg.update(0, 0);

  for (int i = 1; i <= n + 1; ++i) {
    long long mn = seg.query_sub(max(i - k, 0), i);
    seg.update(i, mn + a[i]);
  }

  long long ans = 0;
  for (int i = 0; i < n + 2; ++i) ans += a[i];

  ans -= seg.query_sub(n+1, n+2);

  cout << ans << endl;
  return 0;
}
0