structure/convex-hull-trick/persistent-dynamic-li-chao-tree.hpp

• View this file on GitHub
• Last update: 2024-04-27 01:27:28+09:00
• Include: #include "structure/convex-hull-trick/persistent-dynamic-li-chao-tree.hpp"

Code

template <typename T, T x_low, T x_high, T id>
struct PersistentDynamicLiChaoTree {
  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;

  PersistentDynamicLiChaoTree() : root{nullptr} {}

  Node *update(Node *t, Line &x, const T &l, const T &r, const T &x_l,
               const T &x_r) {
    if (!t) return new Node(x);
    auto 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 = new Node(*t);
      t->x = x;
      return t;
    } else {
      T m = (l + r) / 2;
      auto t_m = t->x.get(m), x_m = x.get(m);
      t = new Node(*t);
      if (t_m > x_m) {
        swap(t->x, x);
        if (x_l >= t_l)
          t->l = update(t->l, x, l, m, t_l, t_m);
        else
          t->r = update(t->r, x, m + 1, r, t_m + x.a, t_r);
      } else {
        if (t_l >= x_l)
          t->l = update(t->l, x, l, m, x_l, x_m);
        else
          t->r = update(t->r, x, m + 1, r, x_m + x.a, x_r);
      }
      return t;
    }
  }

  void update(const T &a, const T &b) {
    Line x(a, b);
    root = update(root, x, 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 (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); }
};

#line 1 "structure/convex-hull-trick/persistent-dynamic-li-chao-tree.hpp"
template <typename T, T x_low, T x_high, T id>
struct PersistentDynamicLiChaoTree {
  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;

  PersistentDynamicLiChaoTree() : root{nullptr} {}

  Node *update(Node *t, Line &x, const T &l, const T &r, const T &x_l,
               const T &x_r) {
    if (!t) return new Node(x);
    auto 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 = new Node(*t);
      t->x = x;
      return t;
    } else {
      T m = (l + r) / 2;
      auto t_m = t->x.get(m), x_m = x.get(m);
      t = new Node(*t);
      if (t_m > x_m) {
        swap(t->x, x);
        if (x_l >= t_l)
          t->l = update(t->l, x, l, m, t_l, t_m);
        else
          t->r = update(t->r, x, m + 1, r, t_m + x.a, t_r);
      } else {
        if (t_l >= x_l)
          t->l = update(t->l, x, l, m, x_l, x_m);
        else
          t->r = update(t->r, x, m + 1, r, x_m + x.a, x_r);
      }
      return t;
    }
  }

  void update(const T &a, const T &b) {
    Line x(a, b);
    root = update(root, x, 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 (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); }
};

Back to top page