ac-library-rb 0.5.0 → 0.6.0

data/document_en/two_sat.md

@@ -17,7 +17,7 @@ it decides whether there is a truth assignment that satisfies all clauses.
 
 ## Class Methods
 
-### new(n = 0) -> TwoSAT
+### new(n) -> TwoSAT
 
 ```ruby
 ts = TwoSAT.new(n)

data/document_ja/dsu.md

@@ -30,7 +30,7 @@ p d.size(2)     # => 2
 
 ## 特異メソッド
 
-### new(n = 0) -> DSU
+### new(n) -> DSU
 
 ```rb
 d = DSU.new(n)
@@ -48,9 +48,7 @@ d = DSU.new(n)
 **エイリアス**
 
 - `DSU`
-- `DisjointSetUnion`
 - `UnionFind`
-- `UnionFindTree`
 
 ## インスタンスメソッド
 

data/document_ja/lazy_segtree.md

@@ -8,10 +8,10 @@
 
 ## 特異メソッド
 
-### new(v, e, id, op, mapping, composition)
+### new(v, e, id){  }
 
 ```ruby
-seg = LazySegtree.new(v, e, id, op, mapping, compositon)
+seg = LazySegtree.new(v, e, id)
 ```
 
 第1引数は、`Integer`または`Array`です。
@@ -21,8 +21,22 @@ seg = LazySegtree.new(v, e, id, op, mapping, compositon)
 
 第2引数`e`は、単位元です。ブロックで二項演算`op(x, y)`を定義することで、モノイドを定義する必要があります。
 
+また、インスタンス作成後に、`LazySegtree#set_mapping{ }`と`LazySegment#set_composition{ }`を用い、適切にインスタンス変数にprocを設定する必要があります。
+
 **計算量** `O(n)`
 
+### new(v, op, e, mapping, composition, id)
+### new(v, e, id, op, mapping, composition)
+
+```ruby
+seg = LazySegtree.new(v, op, e, mapping, compositon, id)
+seg = LazySegtree.new(v, e, id, op, mapping, compositon)
+```
+
+前者は、本家ライブラリに合わせた引数の順番。  
+後者は、procを後ろにまとめた引数の順番で、これは非推奨。  
+内部で、第2引数がprocかどうかで、場合分けしています。
+
 ## インスタンスメソッド
 
 ### set(pos, x)
@@ -53,7 +67,7 @@ seg.get(pos)
 seg.prod(l, r)
 ```
 
-`op(a[l], ..., a[r - 1])` を返します。
+`op(a[l], ..., a[r - 1])` を返します。引数は半開区間です。`l == r`のとき、単位元`e`を返します。
 
 **制約** `0 ≦ l ≦ r ≦ n`
 
@@ -65,19 +79,24 @@ seg.prod(l, r)
 seg.all_prod
 ```
 
-`op(a[0], ..., a[n - 1])` を返します。
+`op(a[0], ..., a[n - 1])` を返します。サイズが0のときは、単位元`e`を返します。
 
 **計算量** `O(1)`
 
 ### apply(pos, val)
+### apply(s, r, val)
 
 ```ruby
 seg.apply(pos, val)
+seg.apply(l, r, val)
 ```
 
-本家コードで引数が3つのとき実装について、当ライブラリでは`range_apply`という名称で実装しています。実行時間を計測して問題なさそうなら、`apply`メソッドに2引数と3引数の両方に対応できるようにします。
+1. `a[p] = f(a[p])`
+2. 半開区間`i = l..r`について`a[i] = f(a[i])`
 
-**制約** `0 ≦ pos < n`
+**制約** 
+1. `0 ≦ pos < n` 
+2. `0 ≦ l ≦ r ≦ n`
 
 **計算量** `O(log n)`
 
@@ -87,28 +106,44 @@ seg.apply(pos, val)
 seg.range_apply(l, r, val)
 ```
 
+3引数の`apply`を呼んだときに、内部で呼ばれるメソッド。  
+直接、`range_apply`を呼ぶこともできる。
+
 **制約** `0 ≦ l ≦ r ≦ n`
 
 **計算量** `O(log n)`
 
-### max_right
+### max_right(l){  } -> Integer
 
-[TODO] 暇な誰かが書く。
+LazySegtree上で、二分探索をします。
 
-### min_left
+**制約**  `0 ≦ l ≦ n`
 
-[TODO] コード側の実装ができていないです。
+**計算量** `O(log n)`
+
+### min_left(r){  } -> Integer
+
+LazySegtree上で、二分探索をします。
+
+**制約**  `0 ≦ r ≦ n`
+
+**計算量** `O(log n)`
 
 ## Verified
 
-問題のリンクです。コードはないですが、Verified済みです。
-- [AIZU ONLINE JUDGE DSL\_2\_F RMQ and RUQ](http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=DSL_2_F)
-- [AIZU ONLINE JUDGE DSL\_2\_G RSQ and RAQ](http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=DSL_2_G)
+問題のリンクです。Verified済みです。解答はテストコードをご参考ください。
+- [AIZU ONLINE JUDGE DSL\_2\_F RMQ and RUQ](https://onlinejudge.u-aizu.ac.jp/problems/DSL_2_F) ([旧DSL_2_F](http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=DSL_2_F))
+- [AIZU ONLINE JUDGE DSL\_2\_G RSQ and RAQ](https://onlinejudge.u-aizu.ac.jp/problems/DSL_2_G) ([旧DSL_2_G](http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=DSL_2_G))
+- [AIZU ONLINE JUDGE DSL\_2\_H RMQ and RAQ](https://onlinejudge.u-aizu.ac.jp/problems/DSL_2_H) ([旧DSL_2_H](http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=DSL_2_H))
+- [AIZU ONLINE JUDGE DSL\_2\_I RSQ and RUQ](https://onlinejudge.u-aizu.ac.jp/problems/DSL_2_I) ([旧DSL_2_I](http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=DSL_2_I))
 
 以下の問題は、Rubyでは実行時間が厳しくTLEになりACできてないです。
 - [ALPC: K \- Range Affine Range Sum](https://atcoder.jp/contests/practice2/tasks/practice2_k)
 - [ALPC: L \- Lazy Segment Tree](https://atcoder.jp/contests/practice2/tasks/practice2_l)
 
+下記は、ジャッジにだしてないが、サンプルに正解。`max_right`, `min_left`を使う問題。
+- [Quora Programming Challenge 2021: Skyscraper](https://jonathanirvin.gs/files/division2_3d16774b0423.pdf#page=5)
+
 ## 参考リンク
 
 - 当ライブラリ
@@ -120,16 +155,25 @@ seg.range_apply(l, r, val)
   - [本家ライブラリの実装コード lazysegtree.hpp(GitHub)](https://github.com/atcoder/ac-library/blob/master/atcoder/lazysegtree.hpp)
   - [本家ライブラリのテストコード lazysegtree_test.cpp(GitHub)](https://github.com/atcoder/ac-library/blob/master/test/unittest/lazysegtree_test.cpp)
 - セグメントツリーについて
-  - [セグメント木をソラで書きたいあなたに \- hogecoder](https://tsutaj.hatenablog.com/entry/2017/03/29/204841)
-  - [遅延評価セグメント木をソラで書きたいあなたに \- hogecoder](https://tsutaj.hatenablog.com/entry/2017/03/30/224339)
+  - [2017/3 hogecoder: セグメント木をソラで書きたいあなたに](https://tsutaj.hatenablog.com/entry/2017/03/29/204841)
+  - [2017/3 hogecoder: 遅延評価セグメント木をソラで書きたいあなたに](https://tsutaj.hatenablog.com/entry/2017/03/30/224339)
+  - [2017/7 はまやんはまやんはまやん: 競技プログラミングにおけるセグメントツリー問題まとめ](https://blog.hamayanhamayan.com/entry/2017/07/08/173120)
+  - [2020/2 ageprocpp Qiita: Segment Treeことはじめ【後編】](https://qiita.com/ageprocpp/items/9ea58ac181d31cfdfe02)
 - AtCooderLibrary(ACL)のLazySegtreeについて
-  - [使い方 \- ARMERIA 2020/9/22](https://betrue12.hateblo.jp/entry/2020/09/22/194541)
-  - [チートシート \- ARMERIA 2020/9/23](https://betrue12.hateblo.jp/entry/2020/09/23/005940)
-  - [ACLPC: K–Range Affine Range Sumの解説 \|optのブログ 2020/9/27](https://opt-cp.com/cp/lazysegtree-aclpc-k/)
-  - [ACL 基礎実装例集 \- buyoh\.hateblo\.jp 2020/9/27](https://buyoh.hateblo.jp/entry/2020/09/27/190144)
+  - [2020/9/22 ARMERIA: 使い方](https://betrue12.hateblo.jp/entry/2020/09/22/194541)
+  - [2020/9/23 ARMERIA: チートシート](https://betrue12.hateblo.jp/entry/2020/09/23/005940)
+  - [2020/9/27 optのブログ: ACLPC: K–Range Affine Range Sumの解説](https://opt-cp.com/cp/lazysegtree-aclpc-k/)
+  - [2020/9/27 buyoh.hateblo.jp: ACL 基礎実装例集](https://buyoh.hateblo.jp/entry/2020/09/27/190144)
 
 ## 本家ライブラリとの違い
 
 ### `ceil_pow2`ではなく、`bit_length`
 
 本家C++ライブラリは独自定義の`internal::ceil_pow2`関数を用いてますが、本ライブラリではRuby既存のメソッド`Integer#bit_length`を用いています。そちらの方が計測した結果、高速でした。
+
+## 問題点
+
+### ミュータブルな単位元
+
+本家ACL同様に、初期化の際に配列でも数値でもいいとなっている。  
+しかし、数値で要素数を指定した際に、ミュータブルなクラスでも同一オブジェクトで要素を作ってしまっている。

data/document_ja/max_flow.md

@@ -4,7 +4,7 @@
 
 ## 特異メソッド
 
-### new(n = 0) -> MaxFlow
+### new(n) -> MaxFlow
 
 ```ruby
 graph = Maxflow.new(10)

data/document_ja/min_cost_flow.md

@@ -4,7 +4,7 @@
 
 ## 特異メソッド
 
-### new(n = 0) -> MinCostFlow
+### new(n) -> MinCostFlow
 
 ```ruby
 graph = MinCostFlow.new(10)

data/document_ja/scc.md

@@ -6,7 +6,7 @@ Strongly Connected Components
 
 ## 特異メソッド
 
-### new(n = 0) -> SCC
+### new(n) -> SCC
 
 ```ruby
 graph = SCC.new(6)
@@ -50,8 +50,9 @@ graph.scc
 
 ## Verified
 
-[ALPC: G \- SCC](https://atcoder.jp/contests/practice2/tasks/practice2_g)
-- [2538ms 2020/9/8](https://atcoder.jp/contests/practice2/submissions/16569175)
+- [ALPC: G \- SCC](https://atcoder.jp/contests/practice2/tasks/practice2_g)  
+  - [2538ms 2020/9/8](https://atcoder.jp/contests/practice2/submissions/16569175)  
+- [競プロ典型 90 問: 021 - Come Back in One Piece][https://atcoder.jp/contests/typical90/tasks/typical90_u]
 
 # 参考リンク
 

data/document_ja/segtree.md

@@ -4,7 +4,8 @@
 
 ## 特異メソッド
 
-### new(arg, e, &op) -> Segtree
+### new(arg, e){ |x, y| ... } -> Segtree
+### new(arg, op, e) -> Segtree
 
 ```rb
 seg = Segtree.new(arg, e) { |x, y| ... }
@@ -66,7 +67,7 @@ seg.get(pos)
 seg.prod(l, r)
 ```
 
-`op(a[l], ..., a[r - 1])` を返します。
+`op(a[l], ..., a[r - 1])` を返します。引数は半開区間です。`l == r`のとき、単位元`e`を返します。
 
 **制約**
 
@@ -82,7 +83,7 @@ seg.prod(l, r)
 seg.all_prod
 ```
 
-`op(a[0], ..., a[n - 1])` を返します。
+`op(a[0], ..., a[n - 1])` を返します。空のセグメントツリー、つまりサイズが0のとき、単位元`e`を返します。
 
 **計算量**
 
@@ -94,7 +95,24 @@ seg.all_prod
 seg.max_right(l, &f)
 ```
 
-Segtree上で二分探索をします。
+Segtree上で`l <= r <= n`の範囲で、`f(prod(l, r))`の結果を二分探索をして条件に当てはまる`r`を返します。
+
+以下の条件を両方満たす `r` (`l <= r <= n`)を(いずれか一つ)返します。
+
+- `r = l` もしくは `f(prod(l, r))`が`true`となる`r`
+- `r = n` もしくは `f(prod(l, r + 1))`が`false`となる`r`
+
+`prod(l, r)`は半開区間`[l, r)`であることに注意。
+
+**制約**
+
+- `f`を同じ引数で呼んだとき、返り値は同じ。
+- `f(e)`が`true`
+- `0 ≦ l ≦ n`
+
+**計算量** 
+
+- `O(log n)`
 
 ### min_left(r, &f) -> Integer
 
@@ -102,13 +120,32 @@ Segtree上で二分探索をします。
 seg.min_left(r, &f)
 ```
 
-Segtree上で二分探索をします。
+Segtree上で`0 <= l <= r`の範囲で、`f(prod(l, r))`の結果を二分探索をして条件に当てはまる`l`を返します。
+
+以下の条件を両方満たす `l` (`0 <= l <= r`)を(いずれか一つ)返します。
+
+- `l = r` もしくは `f(prod(l, r))`が`true`となる`l`
+- `l = 0` もしくは `f(prod(l - 1, r))`が`false`となる`l`
+
+`prod(l, r)`は半開区間`[l, r)`であることに注意。
+
+**制約**
+
+- `f`を同じ引数で呼んだとき、返り値は同じ。
+- `f(e)`が`true`
+- `0 ≦ l ≦ n`
+
+**計算量**
+
+- `O(log n)`
 
 ## Verified
 
 - [ALPC: J \- Segment Tree](https://atcoder.jp/contests/practice2/tasks/practice2_j)
+  - [AC Code(884ms) max_right](https://atcoder.jp/contests/practice2/submissions/23196480)
+  - [AC Code(914ms) min_left](https://atcoder.jp/contests/practice2/submissions/23197311)
 
-- [F \- Range Xor Query](https://atcoder.jp/contests/abc185/tasks/abc185_f)
+- [ABC185: F \- Range Xor Query](https://atcoder.jp/contests/abc185/tasks/abc185_f)
   - xorのセグメントツリーの基本的な典型問題です。FenwickTree(BIT)をxorに改造するだけでも解けます。
   - [ACコード(1538ms)](https://atcoder.jp/contests/abc185/submissions/18746817): 通常のSegtree解。
   - [ACコード(821ms)](https://atcoder.jp/contests/abc185/submissions/18769200): FenwickTree(BIT)のxor改造版。
@@ -119,14 +156,18 @@ Segtree上で二分探索をします。
   - [当ライブラリの実装コード segtree.rb](https://github.com/universato/ac-library-rb/blob/master/lib/segtree.rb)
   - [当ライブラリのテストコード segtree.rb](https://github.com/universato/ac-library-rb/blob/master/test/segtree_test.rb)
     - テストコードも具体的な使い方として役に立つかもしれまん。
-- 本家
+- 本家ライブラリ
   - [本家ライブラリのドキュメント segtree.md(GitHub)](https://github.com/atcoder/ac-library/blob/master/document_ja/segtree.md)
   - [本家のドキュメント appendix.md(GitHub)](https://github.com/atcoder/ac-library/blob/master/document_ja/appendix.md)
   - [本家ライブラリの実装コード segtree.hpp(GitHub)](https://github.com/atcoder/ac-library/blob/master/atcoder/segtree.hpp)
   - [本家ライブラリのテストコード segtree_test.cpp(GitHub)](https://github.com/atcoder/ac-library/blob/master/test/unittest/segtree_test.cpp)
   - [本家ライブラリのサンプルコード segtree_practice.cpp(GitHub)](https://github.com/atcoder/ac-library/blob/master/test/example/segtree_practice.cpp)
 - セグメントツリーについて
-  - [セグメント木をソラで書きたいあなたに \- hogecoder](https://tsutaj.hatenablog.com/entry/2017/03/29/204841)
+  - [2017/3 hogecoder: セグメント木をソラで書きたいあなたに](https://tsutaj.hatenablog.com/entry/2017/03/29/204841)
+  - [2017/7 はまやんはまやんはまやん: 競技プログラミングにおけるセグメントツリー問題まとめ](https://blog.hamayanhamayan.com/entry/2017/07/08/173120)
+  - [2017/12 ei1333の日記: ちょっと変わったセグメント木の使い方](https://ei1333.hateblo.jp/entry/2017/12/14/000000)  
+    スライドが二分探索について詳しい
+  - [2020/2 ageprocpp@Qiita: Segment Treeことはじめ【前編](https://qiita.com/ageprocpp/items/f22040a57ad25d04d199)
 
 ## 本家ライブラリとの違い等
 

data/document_ja/two_sat.md

@@ -15,7 +15,7 @@
 
 ## 特異メソッド
 
-### new(n = 0) -> TwoSAT
+### new(n) -> TwoSAT
 
 ```ruby
 ts = TwoSAT.new(n)

data/lib/ac-library-rb/version.rb

@@ -1,3 +1,3 @@
 module AcLibraryRb
-  VERSION = "0.5.0".freeze
+  VERSION = "0.6.0".freeze
 end

data/lib/convolution.rb

@@ -122,3 +122,24 @@ class Convolution
 
   private :batterfly, :batterfly_inv, :bsf, :calc_primitive_root
 end
+
+# [EXPERIMENTAL]
+def convolution(a, b, mod: 998244353, k: 35, z: 99)
+  n = a.size
+  m = b.size
+  return [] if n == 0 || m == 0
+
+  raise ArgumentError if a.min < 0 || b.min < 0
+
+  format = "%0#{k}x" # "%024x"
+  sa = ""
+  sb = ""
+  a.each{ |x| sa << (format % x) }
+  b.each{ |x| sb << (format % x) }
+
+  zero = '0'
+  s = zero * z + ("%x" % (sa.hex * sb.hex))
+  i = -(n + m - 1) * k - 1
+
+  Array.new(n + m - 1){ (s[i + 1..i += k] || zero).hex % mod }
+end

data/lib/core_ext/all.rb

@@ -0,0 +1,11 @@
+class Array
+  def to_fenwick_tree
+    FenwickTree.new(self)
+  end
+  alias to_fetree to_fenwick_tree
+
+  def to_priority_queue
+    PriorityQueue.new(self)
+  end
+  alias to_pq to_priority_queue
+end

data/lib/dsu.rb

@@ -1,12 +1,13 @@
 # Disjoint Set Union
 class DSU
-  def initialize(n = 0)
+  def initialize(n)
+    @n = n
+    @parent_or_size = Array.new(n, -1)
     # root node: -1 * component size
     # otherwise: parent
-    @parent_or_size = Array.new(n, -1)
   end
 
-  attr_accessor :parent_or_size
+  attr_reader :parent_or_size, :n
 
   def merge(a, b)
     x = leader(a)
@@ -25,6 +26,10 @@ class DSU
   alias same? same
 
   def leader(a)
+    unless 0 <= a && a < @n
+      raise ArgumentError.new, "#{a} is out of range (0...#{@n})"
+    end
+
     @parent_or_size[a] < 0 ? a : (@parent_or_size[a] = leader(@parent_or_size[a]))
   end
   alias root leader
@@ -39,6 +44,4 @@ class DSU
   end
 end
 
-UnionFind        = DSU
-UnionFindTree    = DSU
-DisjointSetUnion = DSU
+UnionFind = DSU

data/lib/fenwick_tree.rb

@@ -2,7 +2,7 @@
 class FenwickTree
   attr_reader :data, :size
 
-  def initialize(arg = 0)
+  def initialize(arg = nil)
     case arg
     when Array
       @size = arg.size
@@ -17,7 +17,7 @@ class FenwickTree
       @size = arg
       @data = Array.new(@size + 1, 0)
     else
-      raise ArgumentError
+      raise ArgumentError.new("wrong argument. type is Array or Integer")
     end
   end
 
@@ -29,8 +29,25 @@ class FenwickTree
     end
   end
 
-  def sum(l, r)
-    _sum(r) - _sum(l)
+  # .sum(l, r)  # [l, r)  <- Original
+  # .sum(r)     # [0, r)  <- [Experimental]
+  # .sum(l..r)  # [l, r]  <- [Experimental]
+  def sum(a, b = nil)
+    if b
+      _sum(b) - _sum(a)
+    elsif a.is_a?(Range)
+      l = a.begin
+      l += @size if l < 0
+      if r = a.end
+        r += @size if r < 0
+        r += 1 unless a.exclude_end?
+      else
+        r = @size
+      end
+      _sum(r) - _sum(l)
+    else
+      _sum(a)
+    end
   end
 
   def _sum(i)
@@ -41,8 +58,5 @@ class FenwickTree
     end
     res
   end
+  alias left_sum _sum
 end
-
-FeTree            = FenwickTree
-Fetree            = FenwickTree
-BinaryIndexedTree = FenwickTree