library_for_cpp

This documentation is automatically generated by online-judge-tools/verification-helper

View the Project on GitHub Kazun1998/library_for_cpp

:heavy_check_mark: 木 DP
(Tree/Tree_DP.hpp)

Outline

木上の DP を計算するメソッドを提供する.

Theory

このページでは, $T = (V, E)$ を木とする. また, 以下の記号を定義する.

Tree_DP_from_Root

Tree_DP_from_Root では, $\mathrm{dp}(v)$ が以下の要件をみたす場合の各 $\mathrm{dp}(v)$ を求める.

$X$ を集合, $\alpha \in X, f: X \times V \times V \to X$ とする. このとき,

\[\mathrm{dp}(v) = \begin{cases} \alpha & (v = r) \\ f(\mathrm{dp}(\mathrm{par}(v)), \mathrm{par}(v), v) & (\text{otherwise}) \end{cases}\]

が成り立つときに適用できる.

つまり, $\mathrm{dp}(v)$ は根から $v$ へのパスを $u_0, u_1, \dots, u_k$ としたとき,

\[\alpha , f(u_0, u_1), f(u_1, u_2), \dots, f(u_{k-1}, u_k)\]

から定まる値を求めることができる.

ここで, $f$ について,

Tree_DP_from_Leaf

Tree_DP_from_Leaf では, $\mathrm{dp}(v)$ が以下の要件をみたす場合の各 $\mathrm{dp}(v)$ を求める.

$X$ を集合, $(M, \odot)$ を $e \in M$ を単位元とする可換モノイド, $f: X \times V \times V \to X, g: M \times V \to X$ とする. このとき,

\[\mathrm{dp}(v) = g \left(\bigodot_{w \in \mathrm{ch}(v)} f(\mathrm{dp}(w), v, w), v \right)\]

が成り立つときに適用できる.

これは, $\mathrm{dp}(v)$ は $v$ を含む部分木から定まる値を求めることができる.

Examples

Tree_DP_from_Root

[Library Checker] Tree Diameter

適当な頂点 $r$ からの距離 $\mathrm{dist}(v)$ を求める必要がある. これは

\[\mathrm{dp}(v) = \begin{cases} 0 & (v = r) \\ \mathrm{dist}(\mathrm{par}(v)) + \mathrm{weight}(\mathrm{par}(v), v) & (\text{otherwise}) \end{cases}\]

によって書ける. ただし, $uv \in E$ に対して, $\mathrm{weight}(u, v)$ で辺 $uv$ を表すとする.

このとき,

\[X = \mathbb{N}, \quad f(w, u, v) = w + \mathrm{weight}(u, v), \quad \alpha = 0\]

とすればよい.

Tree_DP_from_Leaf

[yukicoder] No.763 Noelちゃんと木遊び

$T$ を $r \in V$ を根とする根付き木と見なす.

各 $v \in V$ に対して, $x_v, y_v \in \mathbb{N}$ をそれぞれ以下で定める.

このとき, $v \in V$ に対して,

\[x_v = \sum_{w \in \mathrm{ch}(v)} \max(x_w, y_w), \quad y_v = \sum_{w \in \mathrm{ch}(v)} \max(x_w, y_w - 1) + 1\]

をみたす.

$\boldsymbol{z}_v := \begin{pmatrix} x_v \ y_v \end{pmatrix} \in \mathbb{N}^2$ と定めると,

\[\mathbb{z}_v = \sum_{w \in \mathrm{ch}(v)} \begin{pmatrix} \max(x_w, y_w) \\ \max(x_w, y_w - 1) \end{pmatrix} + 1\]

である.

そのため,

\[X = \mathbb{N}^2, \quad (M, \odot) = (\mathbb{N}^2, +), \quad f \left(\begin{pmatrix} x \\ y \end{pmatrix}, u, v \right) = \begin{pmatrix} \max(x, y) \\ \max(x, y - 1) \end{pmatrix}, \quad g \left(\begin{pmatrix} x \\ y \end{pmatrix} \right) = \begin{pmatrix} x \\ y + 1 \end{pmatrix}, \quad e = \begin{pmatrix} 0 \\ 0 \end{pmatrix}\]

とすればよい.

なお, 最終解答は $\max(x_r, y_r)$ である. このように, $\mathrm{dp}(r)$ そのままが最終解答になるとは限らないので注意すること.

Contents

Tree_DP_from_Root

vector<X> Tree_DP_from_Root(Tree &T, function<X(X, int, int)> f, const X alpha)

Tree_DP_from_Leaf

vector<X> Tree_DP_from_Leaf(Tree &T, function<M(X, int, int)> f, function<X(M, int)> g, function<M(M, M)> merge, const M unit)

Depends on

Required by

Verified with

Code

#pragma once

#include "Tree.hpp"

template<typename X>
vector<X> Tree_DP_from_Root(Tree &T, function<X(X, int, int)> f, const X alpha) {
    vector<X> data(T.vector_size());

    data[T.get_root()] = alpha;

    auto dfs = [&](auto self, int x) -> void {
        for (int y: T.get_children(x)) {
            data[y] = f(data[x], x, y);
            self(self, y);
        }
    };

    dfs(dfs, T.get_root());
    return data;
}

template<typename X, typename M>
vector<X> Tree_DP_from_Leaf(Tree &T, function<M(X, int, int)> f, function<X(M, int)> g, function<M(M, M)> merge, const M unit) {
    vector<X> data(T.vector_size());

    auto dfs = [&](auto self, int v) -> void {
        M children_product = unit;
        for (int w: T.get_children(v)) {
            self(self, w);
            children_product = merge(children_product, f(data[w], v, w));
        }
        data[v] = g(children_product, v);
    };

    dfs(dfs, T.get_root());
    return data;
}
#line 2 "Tree/Tree_DP.hpp"

#line 2 "Tree/Tree.hpp"

#line 2 "template/template.hpp"

using namespace std;

// intrinstic
#include <immintrin.h>

#include <algorithm>
#include <array>
#include <bitset>
#include <cassert>
#include <cctype>
#include <cfenv>
#include <cfloat>
#include <chrono>
#include <cinttypes>
#include <climits>
#include <cmath>
#include <complex>
#include <concepts>
#include <cstdarg>
#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <fstream>
#include <functional>
#include <initializer_list>
#include <iomanip>
#include <ios>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <map>
#include <memory>
#include <new>
#include <numeric>
#include <ostream>
#include <optional>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <stack>
#include <streambuf>
#include <string>
#include <tuple>
#include <type_traits>
#include <typeinfo>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>

// utility
#line 2 "template/utility.hpp"

using ll = long long;

// a ← max(a, b) を実行する. a が更新されたら, 返り値が true.
template<typename T, typename U>
inline bool chmax(T &a, const U b){
    return (a < b ? a = b, 1: 0);
}

// a ← min(a, b) を実行する. a が更新されたら, 返り値が true.
template<typename T, typename U>
inline bool chmin(T &a, const U b){
    return (a > b ? a = b, 1: 0);
}

// a の最大値を取得する.
template<typename T>
inline T max(const vector<T> &a){
    if (a.empty()) throw invalid_argument("vector is empty.");

    return *max_element(a.begin(), a.end());
}

// vector<T> a の最小値を取得する.
template<typename T>
inline T min(const vector<T> &a){
    if (a.empty()) throw invalid_argument("vector is empty.");

    return *min_element(a.begin(), a.end());
}

// vector<T> a の最大値のインデックスを取得する.
template<typename T>
inline size_t argmax(const vector<T> &a){
    if (a.empty()) throw std::invalid_argument("vector is empty.");

    return distance(a.begin(), max_element(a.begin(), a.end()));
}

// vector<T> a の最小値のインデックスを取得する.
template<typename T>
inline size_t argmin(const vector<T> &a){
    if (a.empty()) throw invalid_argument("vector is empty.");

    return distance(a.begin(), min_element(a.begin(), a.end()));
}
#line 61 "template/template.hpp"

// math
#line 2 "template/math.hpp"

// 演算子
template<typename T>
T add(const T &x, const T &y) { return x + y; }

template<typename T>
T sub(const T &x, const T &y) { return x - y; }

template<typename T>
T mul(const T &x, const T &y) { return x * y; }

template<typename T>
T neg(const T &x) { return -x; }

template<integral T>
T bitwise_and(const T &x, const T &y) { return x & y; }

template<integral T>
T bitwise_or(const T &x, const T &y) { return x | y; }

template<integral T>
T bitwise_xor(const T &x, const T &y) { return x ^ y; }

// 除算に関する関数

// floor(x / y) を求める.
template<integral T, integral U>
auto div_floor(T x, U y){
    return x / y - ((x % y != 0) && ((x < 0) != (y < 0)));
}

// ceil(x / y) を求める.
template<integral T, integral U>
auto div_ceil(T x, U y){
    return x / y + ((x % y != 0) && ((x < 0) == (y < 0)));
}

// x を y で割った余りを求める.
template<integral T, integral U>
auto safe_mod(T x, U y){
    auto q = div_floor(x, y);
    return x - q * y ;
}

// x を y で割った商と余りを求める.
template<integral T, integral U>
auto divmod(T x, U y){
    auto q = div_floor(x, y);
    return make_pair(q, x - q * y);
}

// 四捨五入を求める.
template<integral T, integral U>
auto round(T x, U y){
    auto [q, r] = divmod(x, y);
    if (y < 0) return (r <= div_floor(y, 2)) ? q + 1 : q;
    return (r >= div_ceil(y, 2)) ? q + 1 : q;
}

// 奇数かどうか判定する.
template<integral T>
bool is_odd(const T &x) { return x % 2 != 0; }

// 偶数かどうか判定する.
template<integral T>
bool is_even(const T &x) { return x % 2 == 0; }

// m の倍数かどうか判定する.
template<integral T, integral U>
bool is_multiple(const T &x, const U &m) { return x % m == 0; }

// 正かどうか判定する.
template<typename T>
bool is_positive(const T &x) { return x > 0; }

// 負かどうか判定する.
template<typename T>
bool is_negative(const T &x) { return x < 0; }

// ゼロかどうか判定する.
template<typename T>
bool is_zero(const T &x) { return x == 0; }

// 非負かどうか判定する.
template<typename T>
bool is_non_negative(const T &x) { return x >= 0; }

// 非正かどうか判定する.
template<typename T>
bool is_non_positive(const T &x) { return x <= 0; }

// 指数に関する関数

// x の y 乗を求める.
ll intpow(ll x, ll y){
    ll a = 1;
    while (y){
        if (y & 1) { a *= x; }
        x *= x;
        y >>= 1;
    }
    return a;
}

// x の y 乗を z で割った余りを求める.
template<typename T, integral U>
T modpow(T x, U y, T z) {
    T a = 1;
    while (y) {
        if (y & 1) { (a *= x) %= z; }

        (x *= x) %= z;
        y >>= 1;
    }

    return a;
}

template<typename T>
T sum(const vector<T> &X) {
    T y = T(0);
    for (auto &&x: X) { y += x; }
    return y;
}

template<typename T>
T gcd(const T x, const T y) {
    return y == 0 ? x : gcd(y, x % y);
}

// a x + b y = gcd(a, b) を満たす整数の組 (a, b) に対して, (x, y, gcd(a, b)) を求める.
template<integral T>
tuple<T, T, T> Extended_Euclid(T a, T b) {
    T s = 1, t = 0, u = 0, v = 1;
    while (b) {
        auto [q, r] = divmod(a, b);
        a = b;
        b = r;
        tie(s, t) = make_pair(t, s - q * t);
        tie(u, v) = make_pair(v, u - q * v);
    }

    return make_tuple(s, u, a);
}

// floor(sqrt(N)) を求める (N < 0 のときは, 0 とする).
ll isqrt(const ll &N) { 
    if (N <= 0) { return 0; }

    ll x = sqrtl(N);
    while ((x + 1) * (x + 1) <= N) { x++; }
    while (x * x > N) { x--; }

    return x;
}

// floor(sqrt(N)) を求める (N < 0 のときは, 0 とする).
ll floor_sqrt(const ll &N) { return isqrt(N); }

// ceil(sqrt(N)) を求める (N < 0 のときは, 0 とする).
ll ceil_sqrt(const ll &N) {
    ll x = isqrt(N);
    return x * x == N ? x : x + 1;
}
#line 64 "template/template.hpp"

// inout
#line 1 "template/inout.hpp"
// 入出力
template<class... T>
void input(T&... a){ (cin >> ... >> a); }

void print(){ cout << "\n"; }

template<class T, class... Ts>
void print(const T& a, const Ts&... b){
    cout << a;
    (cout << ... << (cout << " ", b));
    cout << "\n";
}

template<typename T, typename U>
istream &operator>>(istream &is, pair<T, U> &P){
    is >> P.first >> P.second;
    return is;
}

template<typename T, typename U>
ostream &operator<<(ostream &os, const pair<T, U> &P){
    os << P.first << " " << P.second;
    return os;
}

template<typename T>
vector<T> vector_input(int N, int index){
    vector<T> X(N+index);
    for (int i=index; i<index+N; i++) cin >> X[i];
    return X;
}

template<typename T>
istream &operator>>(istream &is, vector<T> &X){
    for (auto &x: X) { is >> x; }
    return is;
}

template<typename T>
ostream &operator<<(ostream &os, const vector<T> &X){
    int s = (int)X.size();
    for (int i = 0; i < s; i++) { os << (i ? " " : "") << X[i]; }
    return os;
}

template<typename T>
ostream &operator<<(ostream &os, const unordered_set<T> &S){
    int i = 0;
    for (T a: S) {os << (i ? " ": "") << a; i++;}
    return os;
}

template<typename T>
ostream &operator<<(ostream &os, const set<T> &S){
    int i = 0;
    for (T a: S) { os << (i ? " ": "") << a; i++; }
    return os;
}

template<typename T>
ostream &operator<<(ostream &os, const unordered_multiset<T> &S){
    int i = 0;
    for (T a: S) { os << (i ? " ": "") << a; i++; }
    return os;
}

template<typename T>
ostream &operator<<(ostream &os, const multiset<T> &S){
    int i = 0;
    for (T a: S) { os << (i ? " ": "") << a; i++; }
    return os;
}

template<typename T>
std::vector<T> input_vector(size_t n, size_t offset = 0) {
    std::vector<T> res;
    // 最初に必要な全容量を確保（再確保を防ぐ）
    res.reserve(n + offset);
    // offset 分をデフォルト値で埋める（特別 indexed 用）
    res.assign(offset, T());
    
    for (size_t i = 0; i < n; ++i) {
        T el;
        if (!(std::cin >> el)) break;
        res.push_back(std::move(el));
    }
    return res;
}
#line 67 "template/template.hpp"

// macro
#line 2 "template/macro.hpp"

// マクロの定義
#define all(x) x.begin(), x.end()
#define len(x) ll(x.size())
#define elif else if
#define unless(cond) if (!(cond))
#define until(cond) while (!(cond))
#define loop while (true)

// オーバーロードマクロ
#define overload2(_1, _2, name, ...) name
#define overload3(_1, _2, _3, name, ...) name
#define overload4(_1, _2, _3, _4, name, ...) name
#define overload5(_1, _2, _3, _4, _5, name, ...) name

// 繰り返し系
#define rep1(n) for (ll i = 0; i < n; i++)
#define rep2(i, n) for (ll i = 0; i < n; i++)
#define rep3(i, a, b) for (ll i = a; i < b; i++)
#define rep4(i, a, b, c) for (ll i = a; i < b; i += c)
#define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__)

#define foreach1(x, a) for (auto &&x: a)
#define foreach2(x, y, a) for (auto &&[x, y]: a)
#define foreach3(x, y, z, a) for (auto &&[x, y, z]: a)
#define foreach4(x, y, z, w, a) for (auto &&[x, y, z, w]: a)
#define foreach(...) overload5(__VA_ARGS__, foreach4, foreach3, foreach2, foreach1)(__VA_ARGS__)
#line 70 "template/template.hpp"

// bitop
#line 2 "template/bitop.hpp"

// 非負整数 x の bit legnth を求める.
ll bit_length(ll x) {
    if (x == 0) { return 0; }
    return (sizeof(long) * CHAR_BIT) - __builtin_clzll(x);
}

// 非負整数 x の popcount を求める.
ll popcount(ll x) { return __builtin_popcountll(x); }

// 正の整数 x に対して, floor(log2(x)) を求める.
ll floor_log2(ll x) { return bit_length(x) - 1; }

// 正の整数 x に対して, ceil(log2(x)) を求める.
ll ceil_log2(ll x) { return bit_length(x - 1); }

// x の第 k ビットを取得する
int get_bit(ll x, int k) { return (x >> k) & 1; }

// x のビット列を取得する.
// k はビット列の長さとする.
vector<int> get_bits(ll x, int k) {
    vector<int> bits(k);
    rep(i, k) {
        bits[i] = x & 1;
        x >>= 1;
    }

    return bits;
}

// x のビット列を取得する.
vector<int> get_bits(ll x) { return get_bits(x, bit_length(x)); }
#line 73 "template/template.hpp"

// exception
#line 2 "template/exception.hpp"

class NotExist: public exception {
    private:
    string message;

    public:
    NotExist() : message("求めようとしていたものは存在しません.") {}

    const char* what() const noexcept override {
        return message.c_str();
    }
};
#line 4 "Tree/Tree.hpp"

class Tree {
    private:
    int N, _offset, root;
    vector<int> parent;
    vector<vector<int>> children;

    int N_bit;
    bool locked;

    public:
    Tree(int N, int _offset = 0): N(N), _offset(_offset), N_bit(0) {
        parent.assign(N + _offset, -1);
        for (; (1 << N_bit) <= N; N_bit++) {}
        locked = false;
    }

    bool is_locked() const { return locked; }

    public:
    inline void set_root(const int &x) {
        assert (!is_locked());
        root = x;
    }

    inline int vector_size() const { return order() + offset(); }

    inline int get_root() const { return root; }
    inline int get_parent(const int &x) const { return parent[x]; }
    inline vector<int> get_children(const int &x) const { return children[x]; }

    public:
    // 頂点 x の親を頂点 y に設定する.
    inline void set_parent(const int &x, const int &y) {
        assert (!is_locked());
        parent[x] = y;
    }

    // 頂点 x の子の一つに頂点 y を設定する.
    inline void set_child(const int &x, const int &y) { set_parent(y, x); }

    // 木を確定させる
    void seal() {
        assert(!is_locked());

        parent[root] = -1;
        children.assign(N + offset(), vector<int>());
        for (int v = offset(); v < N + offset(); v++) {
            unless(is_root(v)) { children[parent[v]].emplace_back(v); }
        }

        locked = true;
        bfs();
    }

    private:
    vector<int> depth;
    vector<vector<int>> tower;
    void bfs() {
        assert(is_locked());

        tower.assign(N, {});
        depth.assign(N + offset(), -1);

        deque<int> Q{ root };
        tower[0] = { root };
        depth[root] = 0;

        while (!Q.empty()){
            int x = Q.front(); Q.pop_front();

            for (int y: children[x]) {
                depth[y] = depth[x] + 1;
                tower[depth[y]].emplace_back(y);
                Q.push_back(y);
            }
        }
    }

    public:
    vector<int> top_down() const {
        vector<int> res;
        for (auto layer: tower) {
            res.insert(res.end(), layer.begin(), layer.end());
        }

        return res;
    }

    public:
    vector<int> bottom_up() const {
        vector<int> res;
        for (auto it = tower.rbegin(); it != tower.rend(); ++it) {
            const auto &layer = *it;
            res.insert(res.end(), layer.begin(), layer.end());
        }

        return res;
    }

    // 1 頂点に関する情報
    public:

    // x は根?
    bool is_root(const int &x) const { return x == root; }

    // x は葉?
    bool is_leaf(const int &x) const {
        assert(is_locked());
        return children[x].empty();
    }

    // x の次数
    int degree(const int &x) const {
        assert(is_locked());
        int d = children[x].size();
        if (is_root(x)) { d--; }
        return d;
    }

    // 頂点 x の深さを求める.
    inline int vertex_depth(const int &x) { return depth[x]; }

    // 2 頂点に関する条件

    // x は y の親か?
    bool is_parent(const int &x, const int &y) const {
        assert(is_locked());
        return !is_root(y) && x == parent[y];
    }

    // x は y の個か?
    inline bool is_children(const int &x, const int &y) const { return is_parent(y, x); }

    // x と y は兄弟 (親が同じ) か?
    bool is_brother(const int &x, const int &y) const {
        assert(is_locked());
        return !is_root(x) && !is_root(y) && parent[x] == parent[y];
    }

    private:
    bool has_upper_list = false;
    vector<vector<int>> upper_list;

    void build_upper_list() {
        assert(is_locked());

        if (has_upper_list) { return; }

        has_upper_list = true;

        upper_list.assign(N_bit, vector<int>(N + offset(), -1));

        // Step I
        for (int i = offset(); i < N + offset(); i++) {
            if (is_root(i)) { upper_list[0][i] = i; }
            else { upper_list[0][i] = parent[i]; }
        }

        // Step II
        for (int k = 1; k < N_bit; k++) {
            for (int i = offset(); i < N + offset(); i++) {
                upper_list[k][i] = upper_list[k - 1][upper_list[k - 1][i]];
            }
        }
    }

    public:
    // 頂点 x から見て k 代前の頂点を求める.
    // vertex_depth(x) < k のとき返り値は over = true ならば root, false ならば, -1 である.
    int upper(int x, int k, bool over = true) {
        assert(is_locked());

        build_upper_list();
        if (vertex_depth(x) < k) { return over? root: -1; }

        for(int b = 0; k; k >>= 1, b++){ 
            if (k & 1) { x = upper_list[b][x]; }
        }

        return x;
    }

    public:
    // 頂点 x と頂点 y の最小共通先祖を求める.
    int lowest_common_ancestor(int x, int y) {
        assert(is_locked());

        if (vertex_depth(x) > vertex_depth(y)) { swap(x, y); }
        y = upper(y, vertex_depth(y) - vertex_depth(x));

        if (is_root(x) || x == y) { return x; }

        for (int k = N_bit - 1; k >= 0; k--) {
            int px = upper_list[k][x], py = upper_list[k][y];
            if (px != py) { x = px, y = py; }
        }

        return is_root(x) ? root : parent[x];
    }

    int lowest_common_ancestor_greedy(int x, int y) {
        assert(is_locked());

        if (vertex_depth(x) > vertex_depth(y)) { swap(x, y); }

        while (vertex_depth(x) < vertex_depth(y)) {
            y = parent[y];
        }

        while (x != y) {
            x = get_parent(x);
            y = get_parent(y);
        }

        return x;
    }

    // 2 頂点 x, y 間の距離を求める.
    int distance(int x, int y) {
        return vertex_depth(x) + vertex_depth(y) - 2 * vertex_depth(lowest_common_ancestor(x, y));
    }

    private:
    bool has_euler_tour_vertex = false, has_euler_tour_edge = false;

    public:
    vector<int> in_time, out_time;
    vector<int> euler_tour_vertex;
    vector<tuple<int, int, int>> euler_tour_edge;

    // Euler Tour に関する計算を行う.
    void calculate_euler_tour_vertex() {
        if(has_euler_tour_vertex) { return; }

        euler_tour_vertex.clear();
        in_time.assign(N + offset(), -1);
        out_time.assign(N + offset(), -1);

        auto dfs = [&](auto self, int x) -> void {
            in_time[x] = (int)euler_tour_vertex.size();
            euler_tour_vertex.emplace_back(x);

            for (int y: children[x]) {
                self(self, y);
            }

            out_time[x] = (int)euler_tour_vertex.size() - 1;
            unless(is_root(x)) { euler_tour_vertex.emplace_back(parent[x]); }
        };

        dfs(dfs, root);

        has_euler_tour_vertex = true;
    }

    void calculate_euler_tour_edge() {
        if(has_euler_tour_edge) { return; }

        calculate_euler_tour_vertex();
        euler_tour_edge.clear();

        for (int t = 0; t < 2 * (N - 1); t++) {
            int x = euler_tour_vertex[t], y = euler_tour_vertex[t + 1];
            int k = (x == parent[y]) ? 1 : -1;
            euler_tour_edge.emplace_back(make_tuple(x, y, k));
        }

        has_euler_tour_edge = true;
    }

    // 頂点 u から頂点 v へ向かうパスにおいて k 番目 (0-indexed) に通る頂点 (パスの長さが k より大きい場合は over)
    int jump(int u, int v, int k, int over = -1) {
        if (k == 0) { return u; }

        int w = lowest_common_ancestor(u, v);
        int dist_uw = vertex_depth(u) - vertex_depth(w);
        int dist_wv = vertex_depth(v) - vertex_depth(w);
        int dist_uv = dist_uw + dist_wv;

        if (dist_uv < k) { 
            return over;
        } else if (k <= dist_uw) {
            return upper(u, k);
        } else {
            return upper(v, dist_uv - k);
        }
    }

    vector<int> path(int u, int v) {
        int w = lowest_common_ancestor_greedy(u, v);

        vector<int> path_first{u}, path_second{v};

        while (u != w) {
            u = get_parent(u);
            path_first.emplace_back(u);
        }

        while (v != w) {
            v = get_parent(v);
            path_second.emplace_back(v);
        }

        path_second.pop_back();
        reverse(path_second.begin(), path_second.end());

        path_first.insert(path_first.end(), make_move_iterator(path_second.begin()), make_move_iterator(path_second.end()));

        return path_first;
    }

    inline int order() const { return N; }
    inline int offset() const { return _offset; }
};

Tree Construct_Tree(int N, vector<pair<int, int>> edges, int root, int offset = 0) {
    vector<vector<int>> adj(N + offset, vector<int>());
    for (auto &[u, v]: edges) {
        adj[u].emplace_back(v);
        adj[v].emplace_back(u);
    }

    Tree T(N, offset);
    T.set_root(root);

    vector<bool> seen(N + 1, false);
    seen[root] = true;
    vector<int> stack({root});

    until(stack.empty()) {
        int v = stack.back();
        stack.pop_back();

        for (int w: adj[v]) {
            if (seen[w]) { continue; }

            seen[w] = true;
            T.set_parent(w, v);
            stack.emplace_back(w);
        }
    }

    T.seal();
    return T;
}
#line 4 "Tree/Tree_DP.hpp"

template<typename X>
vector<X> Tree_DP_from_Root(Tree &T, function<X(X, int, int)> f, const X alpha) {
    vector<X> data(T.vector_size());

    data[T.get_root()] = alpha;

    auto dfs = [&](auto self, int x) -> void {
        for (int y: T.get_children(x)) {
            data[y] = f(data[x], x, y);
            self(self, y);
        }
    };

    dfs(dfs, T.get_root());
    return data;
}

template<typename X, typename M>
vector<X> Tree_DP_from_Leaf(Tree &T, function<M(X, int, int)> f, function<X(M, int)> g, function<M(M, M)> merge, const M unit) {
    vector<X> data(T.vector_size());

    auto dfs = [&](auto self, int v) -> void {
        M children_product = unit;
        for (int w: T.get_children(v)) {
            self(self, w);
            children_product = merge(children_product, f(data[w], v, w));
        }
        data[v] = g(children_product, v);
    };

    dfs(dfs, T.get_root());
    return data;
}
Back to top page