Modint.factorial / combination / permutation

NASU41 · NASU41 · commit dbad12f1aa52 · 2021-01-21T23:58:19.000+09:00
diff --git a/2SAT/Readme.md b/2SAT/Readme.md
@@ -71,6 +71,14 @@ public void addNand(int x, boolean f, int y, boolean g)
 
 ならし $O(1)$
 
+### set
+
+```java
+public void set(int x, boolean f)
+```
+
+`(x[i] = f)` というクローズを足します。
+
 ### satisfiable
 
 ```java
diff --git a/2SAT/TwoSAT.java b/2SAT/TwoSAT.java
@@ -29,6 +29,10 @@ public void addImplication(int x, boolean f, int y, boolean g) {
     public void addNand(int x, boolean f, int y, boolean g) {
         addClause(x, !f, y, !g);
     }
+    public void set(int x, boolean f){
+        addClause(x, f, x, f);
+    }
+
 
     public boolean satisfiable() {
         hasCalledSatisfiable = true;
diff --git a/Convolution/README.md b/Convolution/README.md
@@ -15,7 +15,7 @@ public static void main(String[] args) {
     int mod = 998244353;
     long[] a = { 1, 2, 3, 4, 5 };
     long[] b = { 6, 7, 8 };
-    // 畳み込みを計算しまqす。a.length + b.length - 1 の配列が帰ります。
+    // 畳み込みを計算します。a.length + b.length - 1 の配列が帰ります。
     long[] ret = Convolution.convolution(a, b, mod);
     System.out.println(Arrays.toString(ret));
 }
diff --git a/DSU/README.md b/DSU/README.md
@@ -8,7 +8,7 @@
 
 を $O(α(N))$ 時間で処理することが出来ます。
 
-また、内部的に各連結成分ごとに代表となる頂点を 11 つ持っています。辺の追加により連結成分がマージされる時、新たな代表元は元の連結成分の代表元のうちどちらかになります。
+また、内部的に各連結成分ごとに代表となる頂点を 1 つ持っています。辺の追加により連結成分がマージされる時、新たな代表元は元の連結成分の代表元のうちどちらかになります。
 
 ## コンストラクタ
 ### DSU
diff --git a/ModInt/ModInt.java b/ModInt/ModInt.java
@@ -1,3 +1,7 @@
+import java.util.*;
+
+import java.util.ArrayList;
+
 /**
  * @verified
  * <ul>
@@ -6,6 +10,7 @@
  * <li> https://atcoder.jp/contests/abc129/tasks/abc129_f : (2 <= M <= 1000000000)
  * <li> https://atcoder.jp/contests/arc050/tasks/arc050_c : (2 <= M <= 1000000000)
  * <li> https://atcoder.jp/contests/arc012/tasks/arc012_4 : (1 <= M <= 1000000007)
+ * <li> https://atcoder.jp/contests/abc042/tasks/arc058_b : (M = 1000000007, combination ver.)
  * </ul>
  */
 class ModIntFactory {
@@ -15,11 +20,15 @@ class ModIntFactory {
     private final boolean usesMontgomery;
     private final ModArithmetic.ModArithmeticMontgomery maMontgomery;
 
+    private ArrayList<Integer> factorial;
+
     public ModIntFactory(int mod) {
         this.ma = ModArithmetic.of(mod);
         this.usesMontgomery = ma instanceof ModArithmetic.ModArithmeticMontgomery;
         this.maMontgomery = usesMontgomery ? (ModArithmetic.ModArithmeticMontgomery) ma : null;
         this.mod = mod;
+
+        this.factorial = new ArrayList<>();
     }
 
     public ModInt create(long value) {
@@ -31,6 +40,30 @@ public ModInt create(long value) {
         }
     }
 
+    private void prepareFactorial(int max){
+        factorial.ensureCapacity(max+1);
+        if(factorial.size()==0) factorial.add(1);
+        for(int i=factorial.size(); i<=max; i++){
+            factorial.add(ma.mul(factorial.get(i-1), i));
+        }
+    }
+
+    public ModInt factorial(int i){
+        prepareFactorial(i);
+        return create(factorial.get(i));
+    }
+
+    public ModInt permutation(int n, int r){
+        if(n < 0 || r < 0 || n < r) return create(0);
+        prepareFactorial(n);
+        return create(ma.div(factorial.get(n), factorial.get(r)));
+    }
+    public ModInt combination(int n, int r){
+        if(n < 0 || r < 0 || n < r) return create(0);
+        prepareFactorial(n);
+        return create(ma.div(factorial.get(n), ma.mul(factorial.get(r),factorial.get(n-r))));
+    }
+
     public int getMod() {
         return mod;
     }
diff --git a/ModInt/README.md b/ModInt/README.md
@@ -61,6 +61,24 @@ public ModInt create(long value)
 値 `value % mod` を持つ `ModInt` を生成します.
 計算量: $O(1)$
 
+```java
+public ModInt factorial(int i)
+```
+
+値 `(i!) % mod` を持つ `ModInt` を生成します.
+
+```java
+public ModInt permutation(int n, int r)
+```
+
+値 `n P r` を持つ `ModInt` を生成します.
+
+```java
+public ModInt combination(int n, int r)
+```
+
+値 `n C r` を持つ `ModInt` を生成します.
+
 ### ModIntFactory$ModInt
 
 #### mod

Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@ public static void main(String[] args) {`
`15`	`15`	`int mod = 998244353;`
`16`	`16`	`long[] a = { 1, 2, 3, 4, 5 };`
`17`	`17`	`long[] b = { 6, 7, 8 };`
`18`		`- // 畳み込みを計算しまqす。a.length + b.length - 1 の配列が帰ります。`
	`18`	`+ // 畳み込みを計算します。a.length + b.length - 1 の配列が帰ります。`
`19`	`19`	`long[] ret = Convolution.convolution(a, b, mod);`
`20`	`20`	`System.out.println(Arrays.toString(ret));`
`21`	`21`	`}`