summaryrefslogtreecommitdiff
path: root/math/chineseRemainder.cpp
blob: 2cf12f70a033ebb5aad0ffe1fe050c290e072a47 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
// Laufzeit: O(n * log(n)), n := Anzahl der Kongruenzen
// Nur für teilerfremde Moduli.
// Berechnet das kleinste, nicht negative x, das die Kongruenzen simultan löst.
// Alle Lösungen sind kongruent zum kgV der Moduli (Produkt, falls alle teilerfremd sind).
struct ChineseRemainder {
	typedef __int128 lll;
	vector<lll> lhs, rhs, module, inv;
	lll M; // Produkt über die Moduli. Kann leicht überlaufen.

	ll g(vector<lll> &vec) {
		lll res = 0;
		for (int i = 0; i < (int)vec.size(); i++) {
			res += (vec[i] * inv[i]) % M;
			res %= M;
		}
		return res;
	}

	// Fügt Kongruenz l * x = r (mod m) hinzu.
	void addEquation(ll l, ll r, ll m) {
		lhs.push_back(l);
		rhs.push_back(r);
		module.push_back(m);
	}

	// Löst das System.
	ll solve() {
		M = accumulate(module.begin(), module.end(), lll(1), multiplies<lll>());
		inv.resize(lhs.size());
		for (int i = 0; i < (int)lhs.size(); i++) {
			lll x = (M / module[i]) % module[i];
			inv[i] = (multInvers(x, module[i]) * (M / module[i]));
		}
		return (multInvers(g(lhs), M) * g(rhs)) % M;
	}
};