#874918 (C++17) No.2332 Make a Sequence

提出ソース

結果

問題	No.2332 Make a Sequence
ユーザー	Kude
提出日時	2023-05-28 15:25:09
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	235 ms / 2,000 ms
コード長	4,126 bytes
コンパイル時間	4,148 ms
コンパイル使用メモリ	228,528 KB
実行使用メモリ	27,900 KB
最終ジャッジ日時	2023-08-27 12:07:40
合計ジャッジ時間	16,697 ms
ジャッジサーバーID （参考情報）	judge11 / judge14

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 4,376 KB
testcase_01	AC	2 ms 4,376 KB
testcase_02	AC	2 ms 4,380 KB
testcase_03	AC	2 ms 4,380 KB
testcase_04	AC	1 ms 4,376 KB
testcase_05	AC	2 ms 4,380 KB
testcase_06	AC	2 ms 4,376 KB
testcase_07	AC	150 ms 23,668 KB
testcase_08	AC	26 ms 5,916 KB
testcase_09	AC	112 ms 18,264 KB
testcase_10	AC	140 ms 22,928 KB
testcase_11	AC	91 ms 16,068 KB
testcase_12	AC	179 ms 27,620 KB
testcase_13	AC	179 ms 27,784 KB
testcase_14	AC	179 ms 27,604 KB
testcase_15	AC	179 ms 27,788 KB
testcase_16	AC	180 ms 27,604 KB
testcase_17	AC	178 ms 27,660 KB
testcase_18	AC	178 ms 27,612 KB
testcase_19	AC	180 ms 27,652 KB
testcase_20	AC	178 ms 27,656 KB
testcase_21	AC	179 ms 27,900 KB
testcase_22	AC	179 ms 27,532 KB
testcase_23	AC	180 ms 27,660 KB
testcase_24	AC	178 ms 27,536 KB
testcase_25	AC	179 ms 27,564 KB
testcase_26	AC	179 ms 27,532 KB
testcase_27	AC	178 ms 27,160 KB
testcase_28	AC	177 ms 27,408 KB
testcase_29	AC	178 ms 27,220 KB
testcase_30	AC	178 ms 27,152 KB
testcase_31	AC	178 ms 27,280 KB
testcase_32	AC	180 ms 25,876 KB
testcase_33	AC	177 ms 25,872 KB
testcase_34	AC	176 ms 25,872 KB
testcase_35	AC	176 ms 25,872 KB
testcase_36	AC	176 ms 25,900 KB
testcase_37	AC	179 ms 26,996 KB
testcase_38	AC	177 ms 26,908 KB
testcase_39	AC	178 ms 26,884 KB
testcase_40	AC	178 ms 27,144 KB
testcase_41	AC	179 ms 27,008 KB
testcase_42	AC	180 ms 25,652 KB
testcase_43	AC	180 ms 25,636 KB
testcase_44	AC	180 ms 25,792 KB
testcase_45	AC	182 ms 25,988 KB
testcase_46	AC	180 ms 25,740 KB
testcase_47	AC	233 ms 24,416 KB
testcase_48	AC	233 ms 24,408 KB
testcase_49	AC	233 ms 24,384 KB
testcase_50	AC	235 ms 24,368 KB
testcase_51	AC	234 ms 24,368 KB
testcase_52	AC	178 ms 27,072 KB
testcase_53	AC	178 ms 27,144 KB
testcase_54	AC	177 ms 27,136 KB
testcase_55	AC	179 ms 27,120 KB
testcase_56	AC	182 ms 27,088 KB
testcase_57	AC	81 ms 10,400 KB
testcase_58	AC	73 ms 8,984 KB
testcase_59	AC	75 ms 9,260 KB
testcase_60	AC	77 ms 9,800 KB
testcase_61	AC	72 ms 8,908 KB
testcase_62	AC	204 ms 25,760 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include<bits/stdc++.h>
namespace {
#pragma GCC diagnostic ignored "-Wunused-function"
#include<atcoder/all>
#pragma GCC diagnostic warning "-Wunused-function"
using namespace std;
using namespace atcoder;
#define rep(i,n) for(int i = 0; i < (int)(n); i++)
#define rrep(i,n) for(int i = (int)(n) - 1; i >= 0; i--)
#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; } else return false; }
template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return true; } else return false; }
using ll = long long;
using P = pair<int,int>;
using VI = vector<int>;
using VVI = vector<VI>;
using VL = vector<ll>;
using VVL = vector<VL>;


// https://ei1333.github.io/library/structure/convex-hull-trick/dynamic-li-chao-tree.hpp


/**
 * @brief Dynamic-Li-Chao-Tree
 * @docs docs/dynamic-li-chao-tree.md
*/
template< typename T, T x_low, T x_high, T id >
struct DynamicLiChaoTree {

  struct Line {
    T a, b;

    Line(T a, T b) : a(a), b(b) {}

    inline T get(T x) const { return a * x + b; }
  };

  struct Node {
    Line x;
    Node *l, *r;

    Node(const Line &x) : x{x}, l{nullptr}, r{nullptr} {}
  };

  Node *root;

  DynamicLiChaoTree() : root{nullptr} {}

  Node *add_line(Node *t, Line &x, const T &l, const T &r, const T &x_l, const T &x_r) {
    if(!t) return new Node(x);

    T t_l = t->x.get(l), t_r = t->x.get(r);

    if(t_l <= x_l && t_r <= x_r) {
      return t;
    } else if(t_l >= x_l && t_r >= x_r) {
      t->x = x;
      return t;
    } else {
      T m = (l + r) / 2;
      if(m == r) --m;
      T t_m = t->x.get(m), x_m = x.get(m);
      if(t_m > x_m) {
        swap(t->x, x);
        if(x_l >= t_l) t->l = add_line(t->l, x, l, m, t_l, t_m);
        else t->r = add_line(t->r, x, m + 1, r, t_m + x.a, t_r);
      } else {
        if(t_l >= x_l) t->l = add_line(t->l, x, l, m, x_l, x_m);
        else t->r = add_line(t->r, x, m + 1, r, x_m + x.a, x_r);
      }
      return t;
    }
  }

  void add_line(const T &a, const T &b) {
    Line x(a, b);
    root = add_line(root, x, x_low, x_high, x.get(x_low), x.get(x_high));
  }

  Node *add_segment(Node *t, Line &x, const T &a, const T &b, const T &l, const T &r, const T &x_l, const T &x_r) {
    if(r < a || b < l) return t;
    if(a <= l && r <= b) {
      Line y{x};
      return add_line(t, y, l, r, x_l, x_r);
    }
    if(t) {
      T t_l = t->x.get(l), t_r = t->x.get(r);
      if(t_l <= x_l && t_r <= x_r) return t;
    } else {
      t = new Node(Line(0, id));
    }
    T m = (l + r) / 2;
    if(m == r) --m;
    T x_m = x.get(m);
    t->l = add_segment(t->l, x, a, b, l, m, x_l, x_m);
    t->r = add_segment(t->r, x, a, b, m + 1, r, x_m + x.a, x_r);
    return t;
  }

  void add_segment(const T &l, const T &r, const T &a, const T &b) {
    Line x(a, b);
    root = add_segment(root, x, l, r - 1, x_low, x_high, x.get(x_low), x.get(x_high));
  }

  T query(const Node *t, const T &l, const T &r, const T &x) const {
    if(!t) return id;
    if(l == r) return t->x.get(x);
    T m = (l + r) / 2;
    if(m == r) --m;
    if(x <= m) return min(t->x.get(x), query(t->l, l, m, x));
    else return min(t->x.get(x), query(t->r, m + 1, r, x));
  }

  T query(const T &x) const {
    return query(root, x_low, x_high, x);
  }
};

} int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  int n, m;
  cin >> n >> m;
  VI a(n), b(m), c(m);
  rep(i, n) cin >> a[i];
  rep(i, m) cin >> b[i];
  rep(i, m) cin >> c[i];
  VI ab(n + m + 1);
  rep(i, n) ab[i] = a[i];
  ab[n] = -1;
  rep(i, m) ab[n + 1 + i] = b[i];
  auto len = z_algorithm(ab);
  len.erase(len.begin(), len.begin() + n + 1);
  // rep(i, m) cout << len[i] << " \n"[i + 1 == m];
  constexpr long long INF = 1002003004005006007;
  DynamicLiChaoTree<ll, 0, 200010, INF> dp;
  dp.add_segment(0, 1, 0, 0);
  for(int i = 0; i <= m; i++) {
    ll v = dp.query(i);
    if (v == INF) {
      cout << -1 << '\n';
      break;
    }
    if (i == m) {
      cout << v << '\n';
      break;
    }
    // ci(x-i) + v
    dp.add_segment(i, i + len[i] + 1, c[i], v - (ll)i * c[i]);
  }
  exit(0);
}

yukicoder

結果

テストケース

ソースコード