convex/slope_trick/slope_super.hpp

• View this file on GitHub
• Last update: 2024-10-19 00:13:45+09:00
• Include: #include "convex/slope_trick/slope_super.hpp"

Depends on

• alg/monoid/add_pair.hpp
• ds/splaytree/splaytree.hpp

Code

#include "ds/splaytree/splaytree.hpp"
#include "alg/monoid/add_pair.hpp"

namespace SLOPE_TRICK_SUPER {
/*
傾きと座標が全部 T.
(x0,y0,a0) / 傾き変化を splay tree で持つ.
末尾には必ず infty が入っているようにする.
(0,10),(1,6),(3,4),(6,7)
->
(x0,y0,a0)=(0,10,-4)
dat = ([1,3],[3,2])

f(x) の計算, (min,argmin) の計算
加法, 畳み込み

加法: easy
f(x) の計算: sum(a), sum(xa) が要る
畳み込み: x->x+c が要る
*/

template <typename T>
struct Node {
  using value_type = pair<T, T>;
  using operator_type = T;
  using np = Node *;
  using Monoid_X = Monoid_Add_Pair<T>;

  np p, l, r;
  bool rev;
  u32 size;
  pair<T, T> x;    // (x,a)
  pair<T, T> prod; // (a sum, xa sum)
  T add_x;

  static void new_node(np n, const pair<T, T> &x) {
    n->p = n->l = n->r = nullptr, n->rev = 0, n->size = 1;
    n->x = x, n->prod = {x.se, x.fi * x.se}, n->add_x = 0;
  }

  void update() {
    size = 1;
    if (l) { size += l->size; }
    if (r) { size += r->size; }
    prod = {x.se, x.fi * x.se};
    if (l) prod = Monoid_X::op(prod, l->prod);
    if (r) prod = Monoid_X::op(prod, r->prod);
  }

  void prop() {
    assert(!rev);
    if (add_x == 0) return;
    if (l) l->x.fi += add_x, l->prod.se += l->prod.fi * add_x, l->add_x += add_x;
    if (r) r->x.fi += add_x, r->prod.se += r->prod.fi * add_x, r->add_x += add_x;
    add_x = 0;
  }

  void apply(T a) { x.fi += a, prod.se += a * prod.fi, add_x += a; }

  // update, prop 以外で呼ばれるものは、splay 後であることが想定されている。
  // したがってその時点で update, prop 済であることを仮定してよい。
  pair<T, T> get() { return x; }
  void set(const pair<T, T> &xx) {
    x = xx;
    update();
  }
};

// 関数は破壊的な変更にされる
template <typename T>
struct Slope_Trick_Super {
  SplayTree<Node<T>> ST;
  using np = Node<T> *;

  struct FUNC {
    np root; // 定義域がこわれていたら root == empty
    T x0, x1, a0, y0;
    int size() { return (root ? root->size : 0); }
  };

  Slope_Trick_Super(int NODES) : ST(NODES) {}

  // (L,R,a,b) : [L,R] で y=ax+b
  FUNC segment_func(T L, T R, T a, T b) { return {nullptr, L, R, a, a * L + b}; }
  FUNC from_points(vc<pair<T, T>> &point) {
    return from_points(len(point), [&](int i) -> pair<T, T> { return point[i]; });
  }
  template <typename F>
  FUNC from_points(int N, F f) {
    vc<T> X(N), Y(N);
    FOR(i, N) tie(X[i], Y[i]) = f(i);
    if (N == 1) return segment_func(X[0], X[0], 0, Y[0]);
    T a0 = (Y[1] - Y[0]) / (X[1] - X[0]);
    T x0 = X[0], x1 = X.back();
    vc<pair<T, T>> dat;
    T a = a0;
    FOR(i, 1, N - 1) {
      T a1 = (Y[i + 1] - Y[i]) / (X[i + 1] - X[i]);
      dat.eb(X[i], a1 - a), a = a1;
    }
    return FUNC{ST.new_node(dat), x0, x1, a0, Y[0]};
  }

  pair<T, T> domain(FUNC &f) { return {f.x0, f.x1}; }
  T eval(FUNC &f, T x) {
    auto [x0, x1] = domain(f);
    if (!(x0 <= x && x <= x1)) return infty<T>;
    auto [l, r] = ST.split_max_right(f.root, [&](auto dat) -> bool { return dat.fi <= x; });
    auto [a_sum, xa_sum] = ST.prod(l);
    f.root = ST.merge(l, r);
    return f.y0 + f.a0 * (x - x0) + a_sum * x - xa_sum;
  }
  FUNC restrict_domain(FUNC &f, T L, T R) {
    auto [x0, x1] = domain(f);
    chmax(L, x0), chmin(R, x1);
    if (L > R) {
      ST.free_subtree(f.root), f.root = nullptr;
      f.root = nullptr;
      f.x0 = infty<T>, f.x1 = -infty<T>;
      return f;
    }
    // まずは右側をちぢめる. R 以上の傾き変化を消してしまえばよい
    auto [l, r] = ST.split_max_right(f.root, [&](auto dat) -> bool { return dat.fi < R; });
    ST.free_subtree(r);
    // 左側をちぢめる.
    tie(l, r) = ST.split_max_right(l, [&](auto dat) -> bool { return dat.fi <= L; });
    auto [a_sum, xa_sum] = ST.prod(l);
    T new_a0 = f.a0 + a_sum;
    T new_y0 = f.y0 + f.a0 * (L - x0) + a_sum * L - xa_sum;
    ST.free_subtree(l);
    f.root = r, f.x0 = L, f.x1 = R, f.a0 = new_a0, f.y0 = new_y0;
    return f;
  }
  FUNC add(FUNC &f, FUNC &g) {
    T x0 = max(f.x0, g.x0);
    T x1 = min(f.x1, g.x1);
    restrict_domain(f, x0, x1), restrict_domain(g, x0, x1);
    if (x0 > x1) return f;
    T y0 = f.y0 + g.y0, a0 = f.a0 + g.a0;

    if (len(f) < len(g)) swap(f, g);
    auto tmp = ST.get_all(g.root);
    ST.free_subtree(g.root);
    f.x0 = x0, f.x1 = x1, f.a0 = a0, f.y0 = y0;
    if (!f.root) {
      f.root = ST.new_node(tmp);
      return f;
    }
    // あとは単に tmp を挿入していけばいい
    auto dfs = [&](auto &dfs, np root, int l, int r) -> void {
      if (l == r) return;
      root->prop();
      T x = root->x.fi;
      // [l,m),[m,r)
      int m = binary_search([&](int i) -> bool { return tmp[i].fi >= x; }, r, l - 1, 0);
      if (l < m) {
        if (!root->l) {
          root->l = ST.new_node({tmp.begin() + l, tmp.begin() + m});
        } else {
          dfs(dfs, root->l, l, m);
        }
        root->l->p = root;
      }
      if (m < r) {
        if (!root->r) {
          root->r = ST.new_node({tmp.begin() + m, tmp.begin() + r});
        } else {
          dfs(dfs, root->r, m, r);
        }
        root->r->p = root;
      }
      root->update();
    };
    dfs(dfs, f.root, 0, len(tmp));
    return f;
  }
  FUNC sum_all(vc<FUNC> &funcs) {
    assert(len(funcs) >= 1);
    T x0 = funcs[0].x0, x1 = funcs[0].x1;
    for (auto &g: funcs) chmax(x0, g.x0), chmin(x1, g.x1);
    if (x0 > x1) {
      for (auto &f: funcs) { ST.free_subtree(f.root); }
      return {nullptr, infty<T>, -infty<T>, 0, 0};
    }
    for (auto &f: funcs) f = restrict_domain(f, x0, x1);
    int idx = 0;
    FOR(i, 1, len(funcs)) if (len(funcs[idx]) < len(funcs[i])) idx = i;
    swap(funcs[idx], funcs.back());
    FUNC f = POP(funcs);
    vc<pair<T, T>> dat;
    for (auto &g: funcs) {
      f.y0 += g.y0, f.a0 += g.a0;
      auto tmp = ST.get_all(g.root);
      concat(dat, tmp);
      ST.free_subtree(g.root);
    }
    sort(all(dat));
    // あとは単に dat を挿入していけばいい
    if (!f.root) {
      f.root = ST.new_node(dat);
      return f;
    }
    auto dfs = [&](auto &dfs, np root, int l, int r) -> void {
      if (l == r) return;
      root->prop();
      T x = root->x.fi;
      // [l,m),[m,r)
      int m = binary_search([&](int i) -> bool { return dat[i].fi >= x; }, r, l - 1, 0);
      if (l < m) {
        if (!root->l) {
          root->l = ST.new_node({dat.begin() + l, dat.begin() + m});
        } else {
          dfs(dfs, root->l, l, m);
        }
        root->l->p = root;
      }
      if (m < r) {
        if (!root->r) {
          root->r = ST.new_node({dat.begin() + m, dat.begin() + r});
        } else {
          dfs(dfs, root->r, m, r);
        }
        root->r->p = root;
      }
      root->update();
    };
    dfs(dfs, f.root, 0, len(dat));
    return f;
  }

  FUNC shift(FUNC &f, T add_x, T add_y) {
    ST.apply(f.root, add_x);
    f.x0 += add_x, f.x1 += add_x, f.y0 += add_y;
    return f;
  }

  // h[z]=min(x+y==z)f(x)+g(y)
  FUNC convolve(FUNC &f, FUNC &g) {
    if (f.x0 > f.x1 || g.x0 > g.x1) { return {nullptr, infty<T>, -infty<T>, 0, 0}; }
    if (len(f) < len(g)) swap(f, g);
    shift(f, g.x0, g.y0), shift(g, -g.x0, -g.y0);
    if (len(g) == 0) { return convolve_segment(f, 0, g.x1, g.a0, 0); }
    auto tmp = ST.get_all(g.root);
    ST.free_subtree(g.root);
    f = convolve_segment(f, 0, tmp[0].fi, g.a0, 0);
    T a = g.a0;
    FOR(i, len(tmp)) {
      T nx = (i + 1 < len(tmp) ? tmp[i + 1].fi : g.x1);
      a += tmp[i].se;
      f = convolve_segment(f, 0, nx - tmp[i].fi, a, 0);
      for (auto &[x, a]: ST.get_all(f.root)) {
        assert(f.x0 <= x && x <= f.x1);
        if (f.root) assert(!f.root->p);
      }
    }
    return f;
  }

  // [x0,x1], y=ax+b
  FUNC convolve_segment(FUNC &f, T x0, T x1, T a, T b) {
    assert(x0 <= x1);
    if (f.x0 > f.x1) { return {nullptr, infty<T>, -infty<T>, 0, 0}; }
    f = shift(f, x0, a * x0 + b);
    T d = x1 - x0;
    if (d == 0) return f;
    // (0,0) から (x1,ax1) への線分をどこかに挿入する
    // 特に x0, y0 はこのままでよい
    if (f.x0 == f.x1) { return {nullptr, f.x0, f.x0 + d, a, f.y0}; }
    // 先頭に挿入できる場合
    if (a <= f.a0) {
      ST.apply(f.root, d);
      f.root = ST.merge(ST.new_node({f.x0 + d, f.a0 - a}), f.root);
      f.x1 += d, f.a0 = a;
      return f;
    }
    // 末尾に挿入できる場合
    T a_last = f.a0 + ST.prod(f.root).fi;
    if (a_last <= a) {
      f.root = ST.merge(f.root, ST.new_node({f.x1, a - a_last}));
      f.x1 += d;
      return f;
    }
    // 間のどこかに挿入
    auto [l, r] = ST.split_max_right_prod(f.root, [&](auto prod) -> bool { return f.a0 + prod.fi < a; });
    T asum = ST.prod(l).fi;
    T a1 = a - (asum + f.a0);
    auto [xx, aa] = ST.get(r, 0);
    ST.apply(r, d);
    ST.set(r, 0, {xx + d, aa - a1});
    f.root = ST.merge3(l, ST.new_node({xx, a1}), r);
    f.x1 += d;
    return f;
  }

  FUNC add_const(FUNC &f, T a) {
    f.y0 += a;
    return f;
  }

  FUNC add_linear(FUNC &f, T a, T b) {
    f.y0 += a * f.x0 + b;
    f.a0 += a;
    return f;
  }

  // (a-x)+
  FUNC add_a_minus_x(FUNC &f, T a) {
    auto [x0, x1] = domain(f);
    if (x0 > x1) return f;
    if (a <= x0) return f;
    if (x1 <= a) return add_linear(f, -1, a);
    vc<pair<T, T>> point;
    point.eb(x0, a - x0), point.eb(a, 0), point.eb(x1, 0);
    FUNC g = from_points(point);
    return add(f, g);
  }

  // (x-a)+
  FUNC add_x_minus_a(FUNC &f, T a) {
    auto [x0, x1] = domain(f);
    if (x0 > x1) return f;
    if (a <= x0) return add_linear(f, 1, -a);
    if (x1 <= a) return f;
    vc<pair<T, T>> point;
    point.eb(x0, 0), point.eb(a, 0), point.eb(x1, x1 - a);
    FUNC g = from_points(point);
    return add(f, g);
  }

  // |x-a|
  FUNC add_abs(FUNC &f, T a) {
    f = add_a_minus_x(f, a);
    f = add_x_minus_a(f, a);
    return f;
  }

  // fx,x
  pair<T, T> get_min(FUNC &f) {
    if (f.x0 > f.x1) return {infty<T>, 0};
    if (f.a0 >= 0) { return {f.y0, f.x0}; }
    auto [l, r] = ST.split_max_right_prod(f.root, [&](auto prod) -> bool { return f.a0 + prod.fi < 0; });
    auto [asum, xasum] = ST.prod(l);
    T x = (r ? ST.get(r, 0).fi : f.x1);
    f.root = ST.merge(l, r);
    T y = f.y0 + f.a0 * (x - f.x0) + x * asum - xasum;
    return {y, x};
  }

  FUNC clear_right(FUNC &f) {
    if (f.a0 >= 0) {
      ST.free_subtree(f.root), f.root = nullptr, f.a0 = 0;
      return f;
    }
    auto [l, r] = ST.split_max_right_prod(f.root, [&](auto prod) -> bool { return f.a0 + prod.fi < 0; });
    f.root = l;
    if (!r) { return f; }
    T x = ST.get(r, 0).fi;
    ST.free_subtree(r);
    f.root = ST.merge(f.root, ST.new_node({x, -(f.a0 + ST.prod(l).fi)}));
    return f;
  }
  FUNC clear_left(FUNC &f) {
    if (f.a0 >= 0) { return f; }
    auto [l, r] = ST.split_max_right_prod(f.root, [&](auto prod) -> bool { return f.a0 + prod.fi < 0; });
    auto [asum, xasum] = ST.prod(l);
    if (!r) {
      // 定数にする
      T x = f.x1;
      T y = f.y0 + f.a0 * (x - f.x0) + x * asum - xasum;
      ST.free_subtree(l);
      f.root = nullptr;
      f.y0 = y, f.a0 = 0;
      return f;
    }
    T x = ST.get(f.root, 0).fi;
    T y = f.y0 + f.a0 * (x - f.x0) + x * asum - xasum;
    T a = f.a0 + asum + ST.get(r, 0).se;
    ST.free_subtree(l);
    f.root = r;
    ST.set(r, 0, {x, a});
    f.y0 = y;
    f.a0 = 0;
    return f;
  }
#ifdef FASTIO
  void debug(FUNC &f) {
    auto dat = ST.get_all(f.root);
    SHOW(f.x0, f.x1, f.a0, f.y0);
    SHOW(dat);
  }
#endif
};
} // namespace SLOPE_TRICK_SUPER
using SLOPE_TRICK_SUPER::Slope_Trick_Super;

#line 2 "ds/splaytree/splaytree.hpp"

/*
update でちゃんと prod が計算されてくれれば prod は op(lprod,x,rprod) でなくてもよい.
*/

// Node 型を別に定義して使う
template <typename Node>
struct SplayTree {
  Node *pool;
  const int NODES;
  int pid;
  using np = Node *;
  using X = typename Node::value_type;
  using A = typename Node::operator_type;
  vc<np> FREE;

  SplayTree(int NODES) : NODES(NODES), pid(0) { pool = new Node[NODES]; }
  ~SplayTree() { delete[] pool; }

  void free_subtree(np c) {
    if (!c) return;
    auto dfs = [&](auto &dfs, np c) -> void {
      if (c->l) dfs(dfs, c->l);
      if (c->r) dfs(dfs, c->r);
      FREE.eb(c);
    };
    dfs(dfs, c);
  }

  void reset() {
    pid = 0;
    FREE.clear();
  }

  np new_root() { return nullptr; }

  np new_node(const X &x) {
    assert(!FREE.empty() || pid < NODES);
    np n = (FREE.empty() ? &(pool[pid++]) : POP(FREE));
    Node::new_node(n, x);
    return n;
  }

  np new_node(const vc<X> &dat) {
    auto dfs = [&](auto &dfs, int l, int r) -> np {
      if (l == r) return nullptr;
      if (r == l + 1) return new_node(dat[l]);
      int m = (l + r) / 2;
      np l_root = dfs(dfs, l, m);
      np r_root = dfs(dfs, m + 1, r);
      np root = new_node(dat[m]);
      root->l = l_root, root->r = r_root;
      if (l_root) l_root->p = root;
      if (r_root) r_root->p = root;
      root->update();
      return root;
    };
    return dfs(dfs, 0, len(dat));
  }

  u32 get_size(np root) { return (root ? root->size : 0); }

  np merge(np l_root, np r_root) {
    if (!l_root) return r_root;
    if (!r_root) return l_root;
    assert((!l_root->p) && (!r_root->p));
    splay_kth(r_root, 0); // splay したので prop 済
    r_root->l = l_root;
    l_root->p = r_root;
    r_root->update();
    return r_root;
  }
  np merge3(np a, np b, np c) { return merge(merge(a, b), c); }
  np merge4(np a, np b, np c, np d) { return merge(merge(merge(a, b), c), d); }

  pair<np, np> split(np root, u32 k) {
    assert(!root || !root->p);
    if (k == 0) return {nullptr, root};
    if (k == (root->size)) return {root, nullptr};
    splay_kth(root, k - 1);
    np right = root->r;
    root->r = nullptr, right->p = nullptr;
    root->update();
    return {root, right};
  }
  tuple<np, np, np> split3(np root, u32 l, u32 r) {
    np nm, nr;
    tie(root, nr) = split(root, r);
    tie(root, nm) = split(root, l);
    return {root, nm, nr};
  }
  tuple<np, np, np, np> split4(np root, u32 i, u32 j, u32 k) {
    np d;
    tie(root, d) = split(root, k);
    auto [a, b, c] = split3(root, i, j);
    return {a, b, c, d};
  }

  // 部分木が区間 [l,r) に対応するようなノードを作って返す
  // そのノードが root になるわけではないので、
  // このノードを参照した後にすぐに splay して根に持ち上げること
  void goto_between(np &root, u32 l, u32 r) {
    if (l == 0 && r == root->size) return;
    if (l == 0) {
      splay_kth(root, r);
      root = root->l;
      return;
    }
    if (r == root->size) {
      splay_kth(root, l - 1);
      root = root->r;
      return;
    }
    splay_kth(root, r);
    np rp = root;
    root = rp->l;
    root->p = nullptr;
    splay_kth(root, l - 1);
    root->p = rp;
    rp->l = root;
    rp->update();
    root = root->r;
  }

  vc<X> get_all(const np &root) {
    vc<X> res;
    auto dfs = [&](auto &dfs, np root) -> void {
      if (!root) return;
      root->prop();
      dfs(dfs, root->l);
      res.eb(root->get());
      dfs(dfs, root->r);
    };
    dfs(dfs, root);
    return res;
  }

  X get(np &root, u32 k) {
    assert(root == nullptr || !root->p);
    splay_kth(root, k);
    return root->get();
  }

  void set(np &root, u32 k, const X &x) {
    assert(root != nullptr && !root->p);
    splay_kth(root, k);
    root->set(x);
  }

  void multiply(np &root, u32 k, const X &x) {
    assert(root != nullptr && !root->p);
    splay_kth(root, k);
    root->multiply(x);
  }

  X prod(np &root, u32 l, u32 r) {
    assert(root == nullptr || !root->p);
    using Mono = typename Node::Monoid_X;
    if (l == r) return Mono::unit();
    assert(0 <= l && l < r && r <= root->size);
    goto_between(root, l, r);
    X res = root->prod;
    splay(root, true);
    return res;
  }

  X prod(np &root) {
    assert(root == nullptr || !root->p);
    using Mono = typename Node::Monoid_X;
    return (root ? root->prod : Mono::unit());
  }

  void apply(np &root, u32 l, u32 r, const A &a) {
    if (l == r) return;
    assert(0 <= l && l < r && r <= root->size);
    goto_between(root, l, r);
    root->apply(a);
    splay(root, true);
  }
  void apply(np &root, const A &a) {
    if (!root) return;
    root->apply(a);
  }

  void reverse(np &root, u32 l, u32 r) {
    assert(root == nullptr || !root->p);
    if (l == r) return;
    assert(0 <= l && l < r && r <= root->size);
    goto_between(root, l, r);
    root->reverse();
    splay(root, true);
  }
  void reverse(np root) {
    if (!root) return;
    root->reverse();
  }

  void rotate(Node *n) {
    // n を根に近づける。prop, update は rotate の外で行う。
    Node *pp, *p, *c;
    p = n->p;
    pp = p->p;
    if (p->l == n) {
      c = n->r;
      n->r = p;
      p->l = c;
    } else {
      c = n->l;
      n->l = p;
      p->r = c;
    }
    if (pp && pp->l == p) pp->l = n;
    if (pp && pp->r == p) pp->r = n;
    n->p = pp;
    p->p = n;
    if (c) c->p = p;
  }

  void prop_from_root(np c) {
    if (!c->p) {
      c->prop();
      return;
    }
    prop_from_root(c->p);
    c->prop();
  }

  void splay(Node *me, bool prop_from_root_done) {
    // これを呼ぶ時点で、me の祖先（me を除く）は既に prop 済であることを仮定
    // 特に、splay 終了時点で me は upd / prop 済である
    if (!prop_from_root_done) prop_from_root(me);
    me->prop();
    while (me->p) {
      np p = me->p;
      np pp = p->p;
      if (!pp) {
        rotate(me);
        p->update();
        break;
      }
      bool same = (p->l == me && pp->l == p) || (p->r == me && pp->r == p);
      if (same) rotate(p), rotate(me);
      if (!same) rotate(me), rotate(me);
      pp->update(), p->update();
    }
    // me の update は最後だけでよい
    me->update();
  }

  void splay_kth(np &root, u32 k) {
    assert(0 <= k && k < (root->size));
    while (1) {
      root->prop();
      u32 sl = (root->l ? root->l->size : 0);
      if (k == sl) break;
      if (k < sl)
        root = root->l;
      else {
        k -= sl + 1;
        root = root->r;
      }
    }
    splay(root, true);
  }

  // check(x), 左側のノード全体が check を満たすように切る
  template <typename F>
  pair<np, np> split_max_right(np root, F check) {
    if (!root) return {nullptr, nullptr};
    assert(!root->p);
    np c = find_max_right(root, check);
    if (!c) {
      splay(root, true);
      return {nullptr, root};
    }
    splay(c, true);
    np right = c->r;
    if (!right) return {c, nullptr};
    right->p = nullptr;
    c->r = nullptr;
    c->update();
    return {c, right};
  }

  // check(x, cnt), 左側のノード全体が check を満たすように切る
  template <typename F>
  pair<np, np> split_max_right_cnt(np root, F check) {
    if (!root) return {nullptr, nullptr};
    assert(!root->p);
    np c = find_max_right_cnt(root, check);
    if (!c) {
      splay(root, true);
      return {nullptr, root};
    }
    splay(c, true);
    np right = c->r;
    if (!right) return {c, nullptr};
    right->p = nullptr;
    c->r = nullptr;
    c->update();
    return {c, right};
  }

  // 左側のノード全体の prod が check を満たすように切る
  template <typename F>
  pair<np, np> split_max_right_prod(np root, F check) {
    if (!root) return {nullptr, nullptr};
    assert(!root->p);
    np c = find_max_right_prod(root, check);
    if (!c) {
      splay(root, true);
      return {nullptr, root};
    }
    splay(c, true);
    np right = c->r;
    if (!right) return {c, nullptr};
    right->p = nullptr;
    c->r = nullptr;
    c->update();
    return {c, right};
  }

  template <typename F>
  np find_max_right(np root, const F &check) {
    // 最後に見つけた ok の点、最後に探索した点
    np last_ok = nullptr, last = nullptr;
    while (root) {
      last = root;
      root->prop();
      if (check(root->x)) {
        last_ok = root;
        root = root->r;
      } else {
        root = root->l;
      }
    }
    splay(last, true);
    return last_ok;
  }

  template <typename F>
  np find_max_right_cnt(np root, const F &check) {
    // 最後に見つけた ok の点、最後に探索した点
    np last_ok = nullptr, last = nullptr;
    ll n = 0;
    while (root) {
      last = root;
      root->prop();
      ll ns = (root->l ? root->l->size : 0);
      if (check(root->x, n + ns + 1)) {
        last_ok = root;
        n += ns + 1;
        root = root->r;
      } else {
        root = root->l;
      }
    }
    splay(last, true);
    return last_ok;
  }

  template <typename F>
  np find_max_right_prod(np root, const F &check) {
    using Mono = typename Node::Monoid_X;
    X prod = Mono::unit();
    // 最後に見つけた ok の点、最後に探索した点
    np last_ok = nullptr, last = nullptr;
    while (root) {
      last = root;
      root->prop();
      np tmp = root->r;
      root->r = nullptr;
      root->update();
      X lprod = Mono::op(prod, root->prod);
      root->r = tmp;
      root->update();
      if (check(lprod)) {
        prod = lprod;
        last_ok = root;
        root = root->r;
      } else {
        root = root->l;
      }
    }
    splay(last, true);
    return last_ok;
  }
};
#line 2 "alg/monoid/add_pair.hpp"

template <typename E>
struct Monoid_Add_Pair {
  using value_type = pair<E, E>;
  using X = value_type;
  static constexpr X op(const X &x, const X &y) {
    return {x.fi + y.fi, x.se + y.se};
  }
  static constexpr X inverse(const X &x) { return {-x.fi, -x.se}; }
  static constexpr X unit() { return {0, 0}; }
  static constexpr bool commute = true;
};
#line 3 "convex/slope_trick/slope_super.hpp"

namespace SLOPE_TRICK_SUPER {
/*
傾きと座標が全部 T.
(x0,y0,a0) / 傾き変化を splay tree で持つ.
末尾には必ず infty が入っているようにする.
(0,10),(1,6),(3,4),(6,7)
->
(x0,y0,a0)=(0,10,-4)
dat = ([1,3],[3,2])

f(x) の計算, (min,argmin) の計算
加法, 畳み込み

加法: easy
f(x) の計算: sum(a), sum(xa) が要る
畳み込み: x->x+c が要る
*/

template <typename T>
struct Node {
  using value_type = pair<T, T>;
  using operator_type = T;
  using np = Node *;
  using Monoid_X = Monoid_Add_Pair<T>;

  np p, l, r;
  bool rev;
  u32 size;
  pair<T, T> x;    // (x,a)
  pair<T, T> prod; // (a sum, xa sum)
  T add_x;

  static void new_node(np n, const pair<T, T> &x) {
    n->p = n->l = n->r = nullptr, n->rev = 0, n->size = 1;
    n->x = x, n->prod = {x.se, x.fi * x.se}, n->add_x = 0;
  }

  void update() {
    size = 1;
    if (l) { size += l->size; }
    if (r) { size += r->size; }
    prod = {x.se, x.fi * x.se};
    if (l) prod = Monoid_X::op(prod, l->prod);
    if (r) prod = Monoid_X::op(prod, r->prod);
  }

  void prop() {
    assert(!rev);
    if (add_x == 0) return;
    if (l) l->x.fi += add_x, l->prod.se += l->prod.fi * add_x, l->add_x += add_x;
    if (r) r->x.fi += add_x, r->prod.se += r->prod.fi * add_x, r->add_x += add_x;
    add_x = 0;
  }

  void apply(T a) { x.fi += a, prod.se += a * prod.fi, add_x += a; }

  // update, prop 以外で呼ばれるものは、splay 後であることが想定されている。
  // したがってその時点で update, prop 済であることを仮定してよい。
  pair<T, T> get() { return x; }
  void set(const pair<T, T> &xx) {
    x = xx;
    update();
  }
};

// 関数は破壊的な変更にされる
template <typename T>
struct Slope_Trick_Super {
  SplayTree<Node<T>> ST;
  using np = Node<T> *;

  struct FUNC {
    np root; // 定義域がこわれていたら root == empty
    T x0, x1, a0, y0;
    int size() { return (root ? root->size : 0); }
  };

  Slope_Trick_Super(int NODES) : ST(NODES) {}

  // (L,R,a,b) : [L,R] で y=ax+b
  FUNC segment_func(T L, T R, T a, T b) { return {nullptr, L, R, a, a * L + b}; }
  FUNC from_points(vc<pair<T, T>> &point) {
    return from_points(len(point), [&](int i) -> pair<T, T> { return point[i]; });
  }
  template <typename F>
  FUNC from_points(int N, F f) {
    vc<T> X(N), Y(N);
    FOR(i, N) tie(X[i], Y[i]) = f(i);
    if (N == 1) return segment_func(X[0], X[0], 0, Y[0]);
    T a0 = (Y[1] - Y[0]) / (X[1] - X[0]);
    T x0 = X[0], x1 = X.back();
    vc<pair<T, T>> dat;
    T a = a0;
    FOR(i, 1, N - 1) {
      T a1 = (Y[i + 1] - Y[i]) / (X[i + 1] - X[i]);
      dat.eb(X[i], a1 - a), a = a1;
    }
    return FUNC{ST.new_node(dat), x0, x1, a0, Y[0]};
  }

  pair<T, T> domain(FUNC &f) { return {f.x0, f.x1}; }
  T eval(FUNC &f, T x) {
    auto [x0, x1] = domain(f);
    if (!(x0 <= x && x <= x1)) return infty<T>;
    auto [l, r] = ST.split_max_right(f.root, [&](auto dat) -> bool { return dat.fi <= x; });
    auto [a_sum, xa_sum] = ST.prod(l);
    f.root = ST.merge(l, r);
    return f.y0 + f.a0 * (x - x0) + a_sum * x - xa_sum;
  }
  FUNC restrict_domain(FUNC &f, T L, T R) {
    auto [x0, x1] = domain(f);
    chmax(L, x0), chmin(R, x1);
    if (L > R) {
      ST.free_subtree(f.root), f.root = nullptr;
      f.root = nullptr;
      f.x0 = infty<T>, f.x1 = -infty<T>;
      return f;
    }
    // まずは右側をちぢめる. R 以上の傾き変化を消してしまえばよい
    auto [l, r] = ST.split_max_right(f.root, [&](auto dat) -> bool { return dat.fi < R; });
    ST.free_subtree(r);
    // 左側をちぢめる.
    tie(l, r) = ST.split_max_right(l, [&](auto dat) -> bool { return dat.fi <= L; });
    auto [a_sum, xa_sum] = ST.prod(l);
    T new_a0 = f.a0 + a_sum;
    T new_y0 = f.y0 + f.a0 * (L - x0) + a_sum * L - xa_sum;
    ST.free_subtree(l);
    f.root = r, f.x0 = L, f.x1 = R, f.a0 = new_a0, f.y0 = new_y0;
    return f;
  }
  FUNC add(FUNC &f, FUNC &g) {
    T x0 = max(f.x0, g.x0);
    T x1 = min(f.x1, g.x1);
    restrict_domain(f, x0, x1), restrict_domain(g, x0, x1);
    if (x0 > x1) return f;
    T y0 = f.y0 + g.y0, a0 = f.a0 + g.a0;

    if (len(f) < len(g)) swap(f, g);
    auto tmp = ST.get_all(g.root);
    ST.free_subtree(g.root);
    f.x0 = x0, f.x1 = x1, f.a0 = a0, f.y0 = y0;
    if (!f.root) {
      f.root = ST.new_node(tmp);
      return f;
    }
    // あとは単に tmp を挿入していけばいい
    auto dfs = [&](auto &dfs, np root, int l, int r) -> void {
      if (l == r) return;
      root->prop();
      T x = root->x.fi;
      // [l,m),[m,r)
      int m = binary_search([&](int i) -> bool { return tmp[i].fi >= x; }, r, l - 1, 0);
      if (l < m) {
        if (!root->l) {
          root->l = ST.new_node({tmp.begin() + l, tmp.begin() + m});
        } else {
          dfs(dfs, root->l, l, m);
        }
        root->l->p = root;
      }
      if (m < r) {
        if (!root->r) {
          root->r = ST.new_node({tmp.begin() + m, tmp.begin() + r});
        } else {
          dfs(dfs, root->r, m, r);
        }
        root->r->p = root;
      }
      root->update();
    };
    dfs(dfs, f.root, 0, len(tmp));
    return f;
  }
  FUNC sum_all(vc<FUNC> &funcs) {
    assert(len(funcs) >= 1);
    T x0 = funcs[0].x0, x1 = funcs[0].x1;
    for (auto &g: funcs) chmax(x0, g.x0), chmin(x1, g.x1);
    if (x0 > x1) {
      for (auto &f: funcs) { ST.free_subtree(f.root); }
      return {nullptr, infty<T>, -infty<T>, 0, 0};
    }
    for (auto &f: funcs) f = restrict_domain(f, x0, x1);
    int idx = 0;
    FOR(i, 1, len(funcs)) if (len(funcs[idx]) < len(funcs[i])) idx = i;
    swap(funcs[idx], funcs.back());
    FUNC f = POP(funcs);
    vc<pair<T, T>> dat;
    for (auto &g: funcs) {
      f.y0 += g.y0, f.a0 += g.a0;
      auto tmp = ST.get_all(g.root);
      concat(dat, tmp);
      ST.free_subtree(g.root);
    }
    sort(all(dat));
    // あとは単に dat を挿入していけばいい
    if (!f.root) {
      f.root = ST.new_node(dat);
      return f;
    }
    auto dfs = [&](auto &dfs, np root, int l, int r) -> void {
      if (l == r) return;
      root->prop();
      T x = root->x.fi;
      // [l,m),[m,r)
      int m = binary_search([&](int i) -> bool { return dat[i].fi >= x; }, r, l - 1, 0);
      if (l < m) {
        if (!root->l) {
          root->l = ST.new_node({dat.begin() + l, dat.begin() + m});
        } else {
          dfs(dfs, root->l, l, m);
        }
        root->l->p = root;
      }
      if (m < r) {
        if (!root->r) {
          root->r = ST.new_node({dat.begin() + m, dat.begin() + r});
        } else {
          dfs(dfs, root->r, m, r);
        }
        root->r->p = root;
      }
      root->update();
    };
    dfs(dfs, f.root, 0, len(dat));
    return f;
  }

  FUNC shift(FUNC &f, T add_x, T add_y) {
    ST.apply(f.root, add_x);
    f.x0 += add_x, f.x1 += add_x, f.y0 += add_y;
    return f;
  }

  // h[z]=min(x+y==z)f(x)+g(y)
  FUNC convolve(FUNC &f, FUNC &g) {
    if (f.x0 > f.x1 || g.x0 > g.x1) { return {nullptr, infty<T>, -infty<T>, 0, 0}; }
    if (len(f) < len(g)) swap(f, g);
    shift(f, g.x0, g.y0), shift(g, -g.x0, -g.y0);
    if (len(g) == 0) { return convolve_segment(f, 0, g.x1, g.a0, 0); }
    auto tmp = ST.get_all(g.root);
    ST.free_subtree(g.root);
    f = convolve_segment(f, 0, tmp[0].fi, g.a0, 0);
    T a = g.a0;
    FOR(i, len(tmp)) {
      T nx = (i + 1 < len(tmp) ? tmp[i + 1].fi : g.x1);
      a += tmp[i].se;
      f = convolve_segment(f, 0, nx - tmp[i].fi, a, 0);
      for (auto &[x, a]: ST.get_all(f.root)) {
        assert(f.x0 <= x && x <= f.x1);
        if (f.root) assert(!f.root->p);
      }
    }
    return f;
  }

  // [x0,x1], y=ax+b
  FUNC convolve_segment(FUNC &f, T x0, T x1, T a, T b) {
    assert(x0 <= x1);
    if (f.x0 > f.x1) { return {nullptr, infty<T>, -infty<T>, 0, 0}; }
    f = shift(f, x0, a * x0 + b);
    T d = x1 - x0;
    if (d == 0) return f;
    // (0,0) から (x1,ax1) への線分をどこかに挿入する
    // 特に x0, y0 はこのままでよい
    if (f.x0 == f.x1) { return {nullptr, f.x0, f.x0 + d, a, f.y0}; }
    // 先頭に挿入できる場合
    if (a <= f.a0) {
      ST.apply(f.root, d);
      f.root = ST.merge(ST.new_node({f.x0 + d, f.a0 - a}), f.root);
      f.x1 += d, f.a0 = a;
      return f;
    }
    // 末尾に挿入できる場合
    T a_last = f.a0 + ST.prod(f.root).fi;
    if (a_last <= a) {
      f.root = ST.merge(f.root, ST.new_node({f.x1, a - a_last}));
      f.x1 += d;
      return f;
    }
    // 間のどこかに挿入
    auto [l, r] = ST.split_max_right_prod(f.root, [&](auto prod) -> bool { return f.a0 + prod.fi < a; });
    T asum = ST.prod(l).fi;
    T a1 = a - (asum + f.a0);
    auto [xx, aa] = ST.get(r, 0);
    ST.apply(r, d);
    ST.set(r, 0, {xx + d, aa - a1});
    f.root = ST.merge3(l, ST.new_node({xx, a1}), r);
    f.x1 += d;
    return f;
  }

  FUNC add_const(FUNC &f, T a) {
    f.y0 += a;
    return f;
  }

  FUNC add_linear(FUNC &f, T a, T b) {
    f.y0 += a * f.x0 + b;
    f.a0 += a;
    return f;
  }

  // (a-x)+
  FUNC add_a_minus_x(FUNC &f, T a) {
    auto [x0, x1] = domain(f);
    if (x0 > x1) return f;
    if (a <= x0) return f;
    if (x1 <= a) return add_linear(f, -1, a);
    vc<pair<T, T>> point;
    point.eb(x0, a - x0), point.eb(a, 0), point.eb(x1, 0);
    FUNC g = from_points(point);
    return add(f, g);
  }

  // (x-a)+
  FUNC add_x_minus_a(FUNC &f, T a) {
    auto [x0, x1] = domain(f);
    if (x0 > x1) return f;
    if (a <= x0) return add_linear(f, 1, -a);
    if (x1 <= a) return f;
    vc<pair<T, T>> point;
    point.eb(x0, 0), point.eb(a, 0), point.eb(x1, x1 - a);
    FUNC g = from_points(point);
    return add(f, g);
  }

  // |x-a|
  FUNC add_abs(FUNC &f, T a) {
    f = add_a_minus_x(f, a);
    f = add_x_minus_a(f, a);
    return f;
  }

  // fx,x
  pair<T, T> get_min(FUNC &f) {
    if (f.x0 > f.x1) return {infty<T>, 0};
    if (f.a0 >= 0) { return {f.y0, f.x0}; }
    auto [l, r] = ST.split_max_right_prod(f.root, [&](auto prod) -> bool { return f.a0 + prod.fi < 0; });
    auto [asum, xasum] = ST.prod(l);
    T x = (r ? ST.get(r, 0).fi : f.x1);
    f.root = ST.merge(l, r);
    T y = f.y0 + f.a0 * (x - f.x0) + x * asum - xasum;
    return {y, x};
  }

  FUNC clear_right(FUNC &f) {
    if (f.a0 >= 0) {
      ST.free_subtree(f.root), f.root = nullptr, f.a0 = 0;
      return f;
    }
    auto [l, r] = ST.split_max_right_prod(f.root, [&](auto prod) -> bool { return f.a0 + prod.fi < 0; });
    f.root = l;
    if (!r) { return f; }
    T x = ST.get(r, 0).fi;
    ST.free_subtree(r);
    f.root = ST.merge(f.root, ST.new_node({x, -(f.a0 + ST.prod(l).fi)}));
    return f;
  }
  FUNC clear_left(FUNC &f) {
    if (f.a0 >= 0) { return f; }
    auto [l, r] = ST.split_max_right_prod(f.root, [&](auto prod) -> bool { return f.a0 + prod.fi < 0; });
    auto [asum, xasum] = ST.prod(l);
    if (!r) {
      // 定数にする
      T x = f.x1;
      T y = f.y0 + f.a0 * (x - f.x0) + x * asum - xasum;
      ST.free_subtree(l);
      f.root = nullptr;
      f.y0 = y, f.a0 = 0;
      return f;
    }
    T x = ST.get(f.root, 0).fi;
    T y = f.y0 + f.a0 * (x - f.x0) + x * asum - xasum;
    T a = f.a0 + asum + ST.get(r, 0).se;
    ST.free_subtree(l);
    f.root = r;
    ST.set(r, 0, {x, a});
    f.y0 = y;
    f.a0 = 0;
    return f;
  }
#ifdef FASTIO
  void debug(FUNC &f) {
    auto dat = ST.get_all(f.root);
    SHOW(f.x0, f.x1, f.a0, f.y0);
    SHOW(dat);
  }
#endif
};
} // namespace SLOPE_TRICK_SUPER
using SLOPE_TRICK_SUPER::Slope_Trick_Super;

Back to top page