/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|  2.66M|  auto operator+(const Affine &t) const -> Affine {
   12|  2.66M|    u32 x = (u64(a) * t.a) % P;
   13|  2.66M|    u32 y = (u64(a) * t.b + b) % P;
   14|  2.66M|    u32 z = (u64(t.a) * c + t.c) % P;
   15|  2.66M|    return {x, y, z};
   16|  2.66M|  }
   17|  66.1k|  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|  1.05M|def reverse(int x) -> void {
   30|  1.05M|  node[x].rev ^= 1;
   31|  1.05M|  std::swap(node[x].aff.b, node[x].aff.c);
   32|  1.05M|}
   33|       |
   34|  2.07M|def pushdown(int x) -> void {
   35|  2.07M|  if (node[x].rev) {
  ------------------
  |  Branch (35:7): [True: 23.06%, False: 76.94%]
  ------------------
   36|   478k|    std::swap(node[x].ch[0], node[x].ch[1]);
   37|   478k|    node[node[x].ch[0]].k = 0;
   38|   478k|    node[node[x].ch[1]].k = 1;
   39|   478k|    reverse(node[x].ch[0]);
   40|   478k|    reverse(node[x].ch[1]);
   41|   478k|    node[x].rev = 0;
   42|   478k|  }
   43|  2.07M|}
   44|       |
   45|  1.33M|def maintain(int x) -> void {
   46|  1.33M|  node[x].aff = node[node[x].ch[0]].aff +
   47|  1.33M|                Affine{node[x].a, node[x].b, node[x].b} +
   48|  1.33M|                node[node[x].ch[1]].aff;
   49|  1.33M|}
   50|       |
   51|  1.33M|def rotateup(int x) -> void {
   52|  1.33M|  int p = node[x].p;
   53|  1.33M|  int g = node[p].p;
   54|  1.33M|  int k = node[x].k;
   55|  1.33M|  int t = node[p].k;
   56|  1.33M|  node[node[x].ch[k ^ 1]].p = p;
   57|  1.33M|  node[node[x].ch[k ^ 1]].k = k;
   58|  1.33M|  node[p].ch[k] = node[x].ch[k ^ 1];
   59|  1.33M|  node[p].p = x;
   60|  1.33M|  node[p].k = k ^ 1;
   61|  1.33M|  node[x].ch[k ^ 1] = p;
   62|  1.33M|  node[x].p = g;
   63|  1.33M|  node[x].k = t;
   64|  1.33M|  if (t != -1) {
  ------------------
  |  Branch (64:7): [True: 22.07%, False: 77.93%]
  ------------------
   65|   293k|    node[g].ch[t] = x;
   66|   293k|  }
   67|  1.33M|  maintain(p);
   68|  1.33M|}
   69|       |
   70|  1.82M|def is_root(int x) -> bool { return node[x].k == -1; }
   71|       |
   72|   737k|def splay(int x) -> void {
   73|   737k|  pushdown(x);
   74|  1.28M|  while (!is_root(x)) {
  ------------------
  |  Branch (74:10): [True: 42.52%, False: 57.48%]
  ------------------
   75|   545k|    if (int p = node[x].p; is_root(p)) {
  ------------------
  |  Branch (75:28): [True: 54.76%, False: 45.24%]
  ------------------
   76|   298k|      pushdown(p);
   77|   298k|      pushdown(x);
   78|   298k|      rotateup(x);
   79|   298k|    } else {
   80|   246k|      int g = node[p].p;
   81|   246k|      pushdown(g);
   82|   246k|      pushdown(p);
   83|   246k|      pushdown(x);
   84|   246k|      if (node[x].k == node[p].k) {
  ------------------
  |  Branch (84:11): [True: 40.59%, False: 59.41%]
  ------------------
   85|   100k|        rotateup(p);
   86|   100k|        rotateup(x);
   87|   146k|      } else {
   88|   146k|        rotateup(x);
   89|   146k|        rotateup(x);
   90|   146k|      }
   91|   246k|    }
   92|   545k|  }
   93|   737k|}
   94|       |
   95|   165k|def access(int x) -> void {
   96|   165k|  splay(x);
   97|   165k|  node[node[x].ch[1]].k = -1;
   98|   165k|  node[x].ch[1] = 0;
   99|   704k|  while (int p = node[x].p) {
  ------------------
  |  Branch (99:14): [True: 76.54%, False: 23.46%]
  ------------------
  100|   539k|    splay(p);
  101|   539k|    node[node[p].ch[1]].k = -1;
  102|   539k|    node[p].ch[1] = x;
  103|   539k|    node[x].k = 1;
  104|   539k|    rotateup(x);
  105|   539k|  }
  106|   165k|}
  107|       |
  108|       |int head[N];
  109|       |struct {
  110|       |  int to;
  111|       |  int next;
  112|       |} edge[N * 2];
  113|       |
  114|  14.8k|def build(int u, int p) -> void {
  115|  44.6k|  for (int e = head[u]; e; e = edge[e].next) {
                                         ^29.7k
  ------------------
  |  Branch (115:25): [True: 66.67%, False: 33.33%]
  ------------------
  116|  29.7k|    int v = edge[e].to;
  117|  29.7k|    if (v != p) {
  ------------------
  |  Branch (117:9): [True: 50.00%, False: 50.00%]
  ------------------
  118|  14.8k|      build(v, u);
  119|  14.8k|      node[v + 1].p = u + 1;
  120|  14.8k|    }
  121|  29.7k|  }
  122|  14.8k|}
  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|  14.8k|  for (int i = 1; i <= n; ++i) {
                                        ^14.8k
  ------------------
  |  Branch (135:19): [True: 99.99%, False: 0.01%]
  ------------------
  136|  14.8k|    node[i] = {.k = -1};
  137|  14.8k|    u32 a = node[i].a = rd.uw();
  138|  14.8k|    u32 b = node[i].b = rd.uw();
  139|  14.8k|    node[i].aff = {a, b, b};
  140|  14.8k|  }
  141|  14.8k|  for (int i = 1; i != n; ++i) {
                                        ^14.8k
  ------------------
  |  Branch (141:19): [True: 99.99%, False: 0.01%]
  ------------------
  142|  14.8k|    int u = rd.uh();
  143|  14.8k|    int v = rd.uh();
  144|  14.8k|    edge[i * 2 | 0] = {v, head[u]}, head[u] = i * 2 | 0;
  145|  14.8k|    edge[i * 2 | 1] = {u, head[v]}, head[v] = i * 2 | 1;
  146|  14.8k|  }
  147|      1|  build(0, 0);
  148|  99.1k|  while (q--) {
  ------------------
  |  Branch (148:10): [True: 100.00%, False: 0.00%]
  ------------------
  149|  99.1k|    let t = rd.u1();
  150|  99.1k|    if (t == 0) {
  ------------------
  |  Branch (150:9): [True: 33.34%, False: 66.66%]
  ------------------
  151|  33.0k|      int u = rd.uh() + 1;
  152|  33.0k|      int v = rd.uh() + 1;
  153|  33.0k|      access(u);
  154|  33.0k|      reverse(u);
  155|  33.0k|      access(v);
  156|  33.0k|      node[node[v].ch[0]].p = 0;
  157|  33.0k|      node[node[v].ch[0]].k = -1;
  158|  33.0k|      node[v].ch[0] = 0;
  159|  33.0k|      int w = rd.uh() + 1;
  160|  33.0k|      int x = rd.uh() + 1;
  161|  33.0k|      access(w);
  162|  33.0k|      reverse(w);
  163|  33.0k|      node[w].p = x;
  164|  33.0k|    }
  165|  99.1k|    if (t == 1) {
  ------------------
  |  Branch (165:9): [True: 33.28%, False: 66.72%]
  ------------------
  166|  32.9k|      int u = rd.uh() + 1;
  167|  32.9k|      node[u].a = rd.uw();
  168|  32.9k|      node[u].b = rd.uw();
  169|  32.9k|      splay(u);
  170|  32.9k|    }
  171|  99.1k|    if (t == 2) {
  ------------------
  |  Branch (171:9): [True: 33.39%, False: 66.61%]
  ------------------
  172|  33.0k|      int u = rd.uh() + 1;
  173|  33.0k|      int v = rd.uh() + 1;
  174|  33.0k|      u32 x = rd.uw();
  175|  33.0k|      access(v);
  176|  33.0k|      reverse(v);
  177|  33.0k|      access(u);
  178|  33.0k|      x = Affine{node[u].a, node[u].b, node[u].b} + x;
  179|  33.0k|      x = node[node[u].ch[0]].aff + x;
  180|  33.0k|      wt.ud(x);
  181|  33.0k|    }
  182|  99.1k|  }
  183|      1|  return 0;
  184|      1|}