/tmp/solutions/build/dynamic_tree_vertex_set_path_composite-main.cpp:
    1|       |#include <common.h>
    2|       |prelude;
    3|       |
    4|       |namespace {
    5|       |
    6|       |constexpr int N = 200001;
    7|       |constexpr u32 P = 998244353;
    8|       |
    9|       |struct Affine {
   10|       |  u32 a, b, c;
   11|  6.98M|  auto operator+(const Affine &t) const -> Affine {
   12|  6.98M|    u32 x = (u64(a) * t.a) % P;
   13|  6.98M|    u32 y = (u64(a) * t.b + b) % P;
   14|  6.98M|    u32 z = (u64(t.a) * c + t.c) % P;
   15|  6.98M|    return {x, y, z};
   16|  6.98M|  }
   17|   133k|  auto operator+(const u32 &t) const -> u32 { return (u64(a) * t + b) % P; }
   18|       |};
   19|       |struct Node {
   20|       |  u32 a;
   21|       |  u32 b;
   22|       |  Affine aff;
   23|       |  int p;
   24|       |  int ch[2];
   25|       |  i8 rev;
   26|       |  i8 k;
   27|       |} node[N];
   28|       |
   29|  2.24M|def reverse(int x) -> void {
   30|  2.24M|  node[x].rev ^= 1;
   31|  2.24M|  std::swap(node[x].aff.b, node[x].aff.c);
   32|  2.24M|}
   33|       |
   34|  5.31M|def pushdown(int x) -> void {
   35|  5.31M|  if (node[x].rev) {
  ------------------
  |  Branch (35:7): [True: 19.27%, False: 80.73%]
  ------------------
   36|  1.02M|    std::swap(node[x].ch[0], node[x].ch[1]);
   37|  1.02M|    node[node[x].ch[0]].k = 0;
   38|  1.02M|    node[node[x].ch[1]].k = 1;
   39|  1.02M|    reverse(node[x].ch[0]);
   40|  1.02M|    reverse(node[x].ch[1]);
   41|  1.02M|    node[x].rev = 0;
   42|  1.02M|  }
   43|  5.31M|}
   44|       |
   45|  3.49M|def maintain(int x) -> void {
   46|  3.49M|  node[x].aff = node[node[x].ch[0]].aff +
   47|  3.49M|                Affine{node[x].a, node[x].b, node[x].b} +
   48|  3.49M|                node[node[x].ch[1]].aff;
   49|  3.49M|}
   50|       |
   51|  3.49M|def rotateup(int x) -> void {
   52|  3.49M|  int p = node[x].p;
   53|  3.49M|  int g = node[p].p;
   54|  3.49M|  int k = node[x].k;
   55|  3.49M|  int t = node[p].k;
   56|  3.49M|  node[node[x].ch[k ^ 1]].p = p;
   57|  3.49M|  node[node[x].ch[k ^ 1]].k = k;
   58|  3.49M|  node[p].ch[k] = node[x].ch[k ^ 1];
   59|  3.49M|  node[p].p = x;
   60|  3.49M|  node[p].k = k ^ 1;
   61|  3.49M|  node[x].ch[k ^ 1] = p;
   62|  3.49M|  node[x].p = g;
   63|  3.49M|  node[x].k = t;
   64|  3.49M|  if (t != -1) {
  ------------------
  |  Branch (64:7): [True: 23.64%, False: 76.36%]
  ------------------
   65|   825k|    node[g].ch[t] = x;
   66|   825k|  }
   67|  3.49M|  maintain(p);
   68|  3.49M|}
   69|       |
   70|  4.63M|def is_root(int x) -> bool { return node[x].k == -1; }
   71|       |
   72|  1.79M|def splay(int x) -> void {
   73|  1.79M|  pushdown(x);
   74|  3.21M|  while (!is_root(x)) {
  ------------------
  |  Branch (74:10): [True: 44.10%, False: 55.90%]
  ------------------
   75|  1.41M|    if (int p = node[x].p; is_root(p)) {
  ------------------
  |  Branch (75:28): [True: 52.12%, False: 47.88%]
  ------------------
   76|   739k|      pushdown(p);
   77|   739k|      pushdown(x);
   78|   739k|      rotateup(x);
   79|   739k|    } else {
   80|   678k|      int g = node[p].p;
   81|   678k|      pushdown(g);
   82|   678k|      pushdown(p);
   83|   678k|      pushdown(x);
   84|   678k|      if (node[x].k == node[p].k) {
  ------------------
  |  Branch (84:11): [True: 42.49%, False: 57.51%]
  ------------------
   85|   288k|        rotateup(p);
   86|   288k|        rotateup(x);
   87|   390k|      } else {
   88|   390k|        rotateup(x);
   89|   390k|        rotateup(x);
   90|   390k|      }
   91|   678k|    }
   92|  1.41M|  }
   93|  1.79M|}
   94|       |
   95|   334k|def access(int x) -> void {
   96|   334k|  splay(x);
   97|   334k|  node[node[x].ch[1]].k = -1;
   98|   334k|  node[x].ch[1] = 0;
   99|  1.73M|  while (int p = node[x].p) {
  ------------------
  |  Branch (99:14): [True: 80.70%, False: 19.30%]
  ------------------
  100|  1.39M|    splay(p);
  101|  1.39M|    node[node[p].ch[1]].k = -1;
  102|  1.39M|    node[p].ch[1] = x;
  103|  1.39M|    node[x].k = 1;
  104|  1.39M|    rotateup(x);
  105|  1.39M|  }
  106|   334k|}
  107|       |
  108|       |int head[N];
  109|       |struct {
  110|       |  int to;
  111|       |  int next;
  112|       |} edge[N * 2];
  113|       |
  114|   200k|def build(int u, int p) -> void {
  115|   599k|  for (int e = head[u]; e; e = edge[e].next) {
                                         ^399k
  ------------------
  |  Branch (115:25): [True: 66.67%, False: 33.33%]
  ------------------
  116|   399k|    int v = edge[e].to;
  117|   399k|    if (v != p) {
  ------------------
  |  Branch (117:9): [True: 50.00%, False: 50.00%]
  ------------------
  118|   199k|      build(v, u);
  119|   199k|      node[v + 1].p = u + 1;
  120|   199k|    }
  121|   399k|  }
  122|   200k|}
  123|       |
  124|       |} // namespace
  125|       |
  126|      1|int main() {
  127|      1|  node[0].aff.a = 1;
  128|      1|  rd rd;
  129|      1|  wt wt;
  130|      1|  int n = rd.uh();
  131|      1|  int q = rd.uh();
  132|      1|#ifdef LOCAL
  133|      1|  std::memset(head, 0, 4 * n);
  134|      1|#endif
  135|   200k|  for (int i = 1; i <= n; ++i) {
                                        ^200k
  ------------------
  |  Branch (135:19): [True: 100.00%, False: 0.00%]
  ------------------
  136|   200k|    node[i] = {.k = -1};
  137|   200k|    u32 a = node[i].a = rd.uw();
  138|   200k|    u32 b = node[i].b = rd.uw();
  139|   200k|    node[i].aff = {a, b, b};
  140|   200k|  }
  141|   200k|  for (int i = 1; i != n; ++i) {
                                        ^199k
  ------------------
  |  Branch (141:19): [True: 100.00%, False: 0.00%]
  ------------------
  142|   199k|    int u = rd.uh();
  143|   199k|    int v = rd.uh();
  144|   199k|    edge[i * 2 | 0] = {v, head[u]}, head[u] = i * 2 | 0;
  145|   199k|    edge[i * 2 | 1] = {u, head[v]}, head[v] = i * 2 | 1;
  146|   199k|  }
  147|      1|  build(0, 0);
  148|   200k|  while (q--) {
  ------------------
  |  Branch (148:10): [True: 100.00%, False: 0.00%]
  ------------------
  149|   200k|    let t = rd.u1();
  150|   200k|    if (t == 0) {
  ------------------
  |  Branch (150:9): [True: 33.51%, False: 66.50%]
  ------------------
  151|  67.0k|      int u = rd.uh() + 1;
  152|  67.0k|      int v = rd.uh() + 1;
  153|  67.0k|      access(u);
  154|  67.0k|      reverse(u);
  155|  67.0k|      access(v);
  156|  67.0k|      node[node[v].ch[0]].p = 0;
  157|  67.0k|      node[node[v].ch[0]].k = -1;
  158|  67.0k|      node[v].ch[0] = 0;
  159|  67.0k|      int w = rd.uh() + 1;
  160|  67.0k|      int x = rd.uh() + 1;
  161|  67.0k|      access(w);
  162|  67.0k|      reverse(w);
  163|  67.0k|      node[w].p = x;
  164|  67.0k|    }
  165|   200k|    if (t == 1) {
  ------------------
  |  Branch (165:9): [True: 33.24%, False: 66.76%]
  ------------------
  166|  66.4k|      int u = rd.uh() + 1;
  167|  66.4k|      node[u].a = rd.uw();
  168|  66.4k|      node[u].b = rd.uw();
  169|  66.4k|      splay(u);
  170|  66.4k|    }
  171|   200k|    if (t == 2) {
  ------------------
  |  Branch (171:9): [True: 33.26%, False: 66.74%]
  ------------------
  172|  66.5k|      int u = rd.uh() + 1;
  173|  66.5k|      int v = rd.uh() + 1;
  174|  66.5k|      u32 x = rd.uw();
  175|  66.5k|      access(v);
  176|  66.5k|      reverse(v);
  177|  66.5k|      access(u);
  178|  66.5k|      x = Affine{node[u].a, node[u].b, node[u].b} + x;
  179|  66.5k|      x = node[node[u].ch[0]].aff + x;
  180|  66.5k|      wt.ud(x);
  181|  66.5k|    }
  182|   200k|  }
  183|      1|  return 0;
  184|      1|}