Loading [MathJax]/jax/output/CommonHTML/jax.js
My Algorithm : kopricky アルゴリズムライブラリ

kopricky アルゴリズムライブラリ

link cut tree

コードについての説明(個人的メモ)

link cut tree とは木の動的な変化(辺の挿入, 辺の削除) の下で連結性やパスクエリなどを効率的に処理するデータ構造である.
木を根から葉に向かうパスに分解し, 各パスを深さをキーとする splay 木で保持するもの. あくまで木に対するデータ構造なので同じ連結成分内の頂点間に辺を貼ったりはできない.
link cut tree はパスクエリを効率良く扱える一方部分木を扱うのにはあまり適していないのでそのようなデータ構造の実装をする際は同じく木の動的連結性を保持するデータ構造である Euler Tour Tree を使ったほうがよい. ただし link cut tree でも更新クエリがなくまた結合則や交換則が成り立つような演算に対しては non-preferred な子の情報を平衡二分木を用いて管理するなどして部分木クエリを実現することは可能である.
アルゴリズムの理解に時間がかかるが, 実装自体はそこまで重くない. 以下ではパス上の頂点に対するクエリ(可換モノイド上の演算であることを仮定)の処理を実現するデータ構造にしているが, reroot を伴わない link, cut, connected だけなら各値の伝播や reverse などが必要ないのでもっと実装は簡潔になる.
パス, 部分木両方のクエリを処理したい場合は "Self-Adjusting Top Trees" [Tarjan, Werneck 2005] を参考にするとよさそう.

時間計算量: 各クエリ ならし O(logn)

コード

  1. // 以下の実装では link, range, query などで reroot が起こるので注意
  2. template<typename T> class LinkCutTree {
  3. public:
  4. struct node {
  5. int sz;
  6. T val, al, lazy;
  7. node *left, *right, *par;
  8. bool rev;
  9. node(const T _val) : sz(1), val(_val), al(_val), lazy(e1),
  10. left(nullptr), right(nullptr), par(nullptr), rev(false){}
  11. bool isRoot() const {
  12. return (!par) || (par->left != this && par->right != this);
  13. }
  14. void push(){
  15. if(lazy != e1){
  16. opr1(val, lazy), al += lazy * sz;
  17. if(left) opr1(left->lazy, lazy);
  18. if(right) opr1(right->lazy, lazy);
  19. lazy = e1;
  20. }
  21. if(!rev) return;
  22. swap(left, right);
  23. if(left) left->rev = !(left->rev);
  24. if(right) right->rev = !(right->rev);
  25. rev = false;
  26. }
  27. void eval(){
  28. sz = 1, al = val;
  29. if(left) left->push(), sz += left->sz, al = opr2(left->al, al);
  30. if(right) right->push(), sz += right->sz, al = opr2(al, right->al);
  31. }
  32. };
  33. static const T e1 = (T)0;
  34. static void opr1(T& arg1, const T arg2){
  35. arg1 += arg2;
  36. }
  37. static T opr2(const T arg1, const T arg2){
  38. return arg1 + arg2;
  39. }
  40.  
  41. private:
  42. // 回転
  43. void rotate(node* u, bool right){
  44. node *p = u->par, *g = p->par;
  45. if(right){
  46. if((p->left = u->right)) u->right->par = p;
  47. u->right = p, p->par = u;
  48. }else{
  49. if((p->right = u->left)) u->left->par = p;
  50. u->left = p, p->par = u;
  51. }
  52. p->eval(), u->eval(), u->par = g;
  53. if(!g) return;
  54. if(g->left == p) g->left = u;
  55. if(g->right == p) g->right = u;
  56. g->eval();
  57. }
  58. // u を splay 木の根にする
  59. void splay(node* u){
  60. while(!(u->isRoot())){
  61. node *p = u->par, *gp = p->par;
  62. if(p->isRoot()){ // zig
  63. p->push(), u->push();
  64. rotate(u, (u == p->left));
  65. }else{
  66. gp->push(), p->push(), u->push();
  67. bool flag = (u == p->left);
  68. if((u == p->left) == (p == gp->left)){ // zig-zig
  69. rotate(p, flag), rotate(u, flag);
  70. }else{ // zig-zag
  71. rotate(u, flag), rotate(u, !flag);
  72. }
  73. }
  74. }
  75. u->push();
  76. }
  77. node* access(node* u){
  78. node* last = nullptr;
  79. for(node* v = u; v; v = v->par){
  80. splay(v);
  81. v->right = last;
  82. v->eval();
  83. last = v;
  84. }
  85. splay(u);
  86. return last;
  87. }
  88. bool connected(node* u, node* v){
  89. access(u), access(v);
  90. return u == v || u->par;
  91. }
  92. void link(node* u, node* v){
  93. evert(u), u->par = v;
  94. }
  95. void cut(node* u){
  96. access(u);
  97. u->left->par = nullptr, u->left = nullptr, u->eval();
  98. }
  99. void cut(node* u, node* v){
  100. access(u), access(v);
  101. if(u->isRoot()) u->par = nullptr;
  102. else v->left->par = nullptr, v->left = nullptr, v->eval();
  103. }
  104. node* lca(node* u, node* v){
  105. access(u);
  106. return access(v);
  107. }
  108. void evert(node* u){
  109. access(u);
  110. u->rev = !(u->rev), u->push();
  111. }
  112. int depth(node* u){
  113. access(u);
  114. return u->sz - 1;
  115. }
  116. void toRoot_range(node* u, const T x){
  117. access(u);
  118. opr1(u->lazy, x), u->push();
  119. }
  120. T toRoot_query(node* u){
  121. access(u);
  122. return u->al;
  123. }
  124. void range(node* u, node* v, const T x){
  125. evert(u), access(v);
  126. opr1(v->lazy, x), v->push();
  127. }
  128. T query(node* u, node* v){
  129. evert(u), access(v);
  130. return v->al;
  131. }
  132.  
  133. public:
  134. const int V;
  135. node** arr;
  136. LinkCutTree(const vector<T>& init_node_value)
  137. : V((int)init_node_value.size()){
  138. arr = new node*[V];
  139. for(int i = 0; i < V; ++i){
  140. arr[i] = new node(init_node_value[i]);
  141. }
  142. }
  143. // ~LinkCutTree(){
  144. // for(int i = 0; i < V; ++i){
  145. // delete arr[i];
  146. // }
  147. // delete[] arr;
  148. // }
  149. // id1 と id2 が同じ木(連結成分)に属するか
  150. bool connected(int id1, int id2){ return connected(arr[id1], arr[id2]); }
  151. // id1 を id2 の non-preferred edge にする
  152. void link(int id1, int id2){ return link(arr[id1], arr[id2]); }
  153. // id とその親の間の辺を削除する
  154. void cut(int id){ return cut(arr[id]); }
  155. // id と id2 の間の辺を削除する
  156. void cut(int id1, int id2){ return cut(arr[id1], arr[id2]); }
  157. // id1 と id2 の LCA を求める
  158. int lca(int id1, int id2){ return static_cast<size_t>(lca(arr[id1], arr[id2]) - arr[0]); }
  159. // 元の木の根を id にする
  160. void evert(int id){ return evert(arr[id]); }
  161. // id の深さを求める
  162. int depth(int id){ return depth(arr[id]); }
  163. // id から根までの間にある頂点すべてに x を足す
  164. void toRoot_range(int id, const T x){ return toRoot_range(arr[id], x); }
  165. // id から根までの間にある頂点すべてのコストの総和を求める
  166. T toRoot_query(int id){ return toRoot_query(arr[id]); }
  167. // id1 と id2 の間にある頂点すべてに x を足す
  168. void range(int id1, int id2, const T x){ return range(arr[id1], arr[id2], x); }
  169. // id1 と id2 の間にある頂点すべてのコストの総和を求める
  170. T query(int id1, int id2){ return query(arr[id1], arr[id2]); }
  171. };

コード(部分木総和クエリを備えたパス加算総和)

  1. template<typename T> class LinkCutTree {
  2. public:
  3. struct node {
  4. int sz, sub1, sub2;
  5. T val, al, lazy, sum1, sum2;
  6. node *left, *right, *par;
  7. bool rev;
  8. node(const T _val) : sz(1), sub1(1), sub2(0), val(_val), al(_val), lazy(0), sum1(_val), sum2(0),
  9. left(nullptr), right(nullptr), par(nullptr), rev(false){}
  10. bool isRoot() const {
  11. return (!par) || (par->left != this && par->right != this);
  12. }
  13. void push(){
  14. if(lazy != 0){
  15. val += lazy, al += lazy * sz;
  16. if(left) left->lazy += lazy;
  17. if(right) right->lazy += lazy;
  18. lazy = 0;
  19. }
  20. if(!rev) return;
  21. swap(left, right);
  22. if(left) left->rev = !(left->rev);
  23. if(right) right->rev = !(right->rev);
  24. rev = false;
  25. }
  26. void eval(){
  27. sz = 1, sub1 = sub2 + 1, al = val, sum1 = sum2 + val;
  28. if(left) left->push(), sz += left->sz, sub1 += left->sub1, al += left->al, sum1 += left->sum1;
  29. if(right) right->push(), sz += right->sz, sub1 += right->sub1, al += right->al, sum1 += right->sum1;
  30. }
  31. };
  32.  
  33. private:
  34. // 回転
  35. void rotate(node* u, bool right){
  36. node *p = u->par, *g = p->par;
  37. if(right){
  38. if((p->left = u->right)) u->right->par = p;
  39. u->right = p, p->par = u;
  40. }else{
  41. if((p->right = u->left)) u->left->par = p;
  42. u->left = p, p->par = u;
  43. }
  44. p->eval(), u->eval(), u->par = g;
  45. if(!g) return;
  46. if(g->left == p) g->left = u;
  47. if(g->right == p) g->right = u;
  48. g->eval();
  49. }
  50. // u を splay 木の根にする
  51. void splay(node* u){
  52. while(!(u->isRoot())){
  53. node *p = u->par, *gp = p->par;
  54. if(p->isRoot()){ // zig
  55. p->push(), u->push();
  56. rotate(u, (u == p->left));
  57. }else{
  58. gp->push(), p->push(), u->push();
  59. bool flag = (u == p->left);
  60. if((u == p->left) == (p == gp->left)){ // zig-zig
  61. rotate(p, flag), rotate(u, flag);
  62. }else{ // zig-zag
  63. rotate(u, flag), rotate(u, !flag);
  64. }
  65. }
  66. }
  67. u->push();
  68. }
  69. node* access(node* u){
  70. node* last = nullptr;
  71. for(node* v = u; v; v = v->par){
  72. splay(v);
  73. if(v->right) v->sub2 += v->right->sub1, v->sum2 += v->right->sum1;
  74. v->right = last;
  75. if(v->right) v->sub2 -= v->right->sub1, v->sum2 -= v->right->sum1;
  76. v->eval();
  77. last = v;
  78. }
  79. splay(u);
  80. return last;
  81. }
  82. bool connected(node* u, node* v){
  83. access(u), access(v);
  84. return u == v || u->par;
  85. }
  86. void link(node* u, node* v){
  87. evert(u), access(v);
  88. u->par = v, v->sub2 += u->sub1, v->sum2 += u->sum1;
  89. }
  90. void cut(node* u){
  91. access(u);
  92. u->left->par = nullptr, u->left = nullptr, u->eval();
  93. }
  94. void cut(node* u, node* v){
  95. access(u), access(v);
  96. if(u->isRoot()) v->sub2 -= u->sub1, v->sum2 -= u->sum1, u->par = nullptr;
  97. else v->left->par = nullptr, v->left = nullptr, v->eval();
  98. }
  99. node* lca(node* u, node* v){
  100. access(u);
  101. return access(v);
  102. }
  103. void evert(node* u){
  104. access(u);
  105. u->rev = !(u->rev), u->push();
  106. }
  107. int depth(node* u){
  108. access(u);
  109. return u->sz - 1;
  110. }
  111. void toRoot_range(node* u, const T x){
  112. access(u);
  113. u->lazy += x, u->push();
  114. }
  115. T toRoot_query(node* u){
  116. access(u);
  117. return u->al;
  118. }
  119. void range(node* u, node* v, const T x){
  120. evert(u), access(v);
  121. v->lazy += x, v->push();
  122. }
  123. T query(node* u, node* v){
  124. evert(u), access(v);
  125. return v->al;
  126. }
  127. int component_size(node* u){
  128. evert(u);
  129. return u->sub1;
  130. }
  131. int subtree_size(node* u){
  132. access(u);
  133. return u->sub2 + 1;
  134. }
  135. T component_query(node* u){
  136. evert(u);
  137. return u->sum1;
  138. }
  139. T subtree_query(node* u){
  140. access(u);
  141. return u->sum2 + u->val;
  142. }
  143.  
  144. public:
  145. const int V;
  146. node** arr;
  147. LinkCutTree(const vector<T>& init_node_value)
  148. : V((int)init_node_value.size()){
  149. arr = new node*[V];
  150. for(int i = 0; i < V; ++i){
  151. arr[i] = new node(init_node_value[i]);
  152. }
  153. }
  154. // ~LinkCutTree(){
  155. // for(int i = 0; i < V; ++i){
  156. // delete arr[i];
  157. // }
  158. // delete[] arr;
  159. // }
  160. // id1 と id2 が同じ木(連結成分)に属するか
  161. bool connected(int id1, int id2){ return connected(arr[id1], arr[id2]); }
  162. // id1 を id2 の non-preferred edge にする
  163. void link(int id1, int id2){ return link(arr[id1], arr[id2]); }
  164. // id とその親の間の辺を削除する
  165. void cut(int id){ return cut(arr[id]); }
  166. // id と id2 の間の辺を削除する
  167. void cut(int id1, int id2){ return cut(arr[id1], arr[id2]); }
  168. // id1 と id2 の LCA を求める
  169. int lca(int id1, int id2){ return static_cast<size_t>(lca(arr[id1], arr[id2]) - arr[0]); }
  170. // 元の木の根を id にする
  171. void evert(int id){ return evert(arr[id]); }
  172. // id の深さを求める
  173. int depth(int id){ return depth(arr[id]); }
  174. // id から根までの間にある頂点すべてに x を足す
  175. void toRoot_range(int id, const T x){ return toRoot_range(arr[id], x); }
  176. // id から根までの間にある頂点すべてのコストの総和を求める
  177. T toRoot_query(int id){ return toRoot_query(arr[id]); }
  178. // id1 と id2 の間にある頂点すべてに x を足す
  179. void range(int id1, int id2, const T x){ return range(arr[id1], arr[id2], x); }
  180. // id1 と id2 の間にある頂点すべてのコストの総和を求める
  181. T query(int id1, int id2){ return query(arr[id1], arr[id2]); }
  182. // id の属する連結成分のサイズを返す
  183. int component_size(int id){ return component_size(arr[id]); }
  184. // id を根とする部分木のサイズを返す
  185. int subtree_size(int id){ return subtree_size(arr[id]); }
  186. // id の属する連結成分内のコストの総和を返す
  187. T component_query(int id){ return component_query(arr[id]); }
  188. // id を根とする部分木内のコストの総和を返す
  189. T subtree_query(int id){ return subtree_query(arr[id]); }
  190. };

verify 用の問題

Atcoder : 白黒ツリー 提出コード
yukicoder : 動的な領主 提出コード
yosupo さんの library checker : Dynamic Tree Vertex Add Subtree Sum 提出コード