-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathmat.cpp
More file actions
101 lines (96 loc) · 2.38 KB
/
mat.cpp
File metadata and controls
101 lines (96 loc) · 2.38 KB
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
typedef long long number;
typedef vector<number> vec;
typedef vector<vec> matrix;
const ll MOD = 1e9+7;
// O( n )
matrix identity(int n) {
matrix A(n, vec(n));
for (int i = 0; i < n; ++i) A[i][i] = 1;
return A;
}
// O( n^3 )
matrix mul(const matrix &A, const matrix &B) {
matrix C(A.size(), vec(B[0].size()));
for (int i = 0; i < (int)C.size(); ++i)
for (int j = 0; j < (int)C[i].size(); ++j)
for (int k = 0; k < (int)A[i].size(); ++k) {
C[i][j] += A[i][k] * B[k][j];
C[i][j] %= MOD;
}
return C;
}
// O( n^3 log e )
matrix pow(const matrix &A, ll e) {
if (e == 0) return identity(A.size());
if (e == 1) return A;
if (e % 2 == 0) {
matrix tmp = pow(A, e/2);
return mul(tmp, tmp);
} else {
matrix tmp = pow(A, e-1);
return mul(A, tmp);
}
}
// O(n)
number tr(const matrix &A) {
number ans = 0;
for (int i = 0; i < (int)A.size(); ++i)
ans += A[i][i];
return ans;
}
// O( n )
number inner_product(const vec &a, const vec &b) {
number ans = 0;
for (int i = 0; i < (int)a.size(); ++i)
ans += a[i] * b[i];
return ans;
}
// O( n^2 )
vec mul(const matrix &A, const vec &x) {
vec y(A.size());
for (int i = 0; i < (int)A.size(); ++i)
for (int j = 0; j < (int)A[0].size(); ++j)
(y[i] += (A[i][j] * x[j] % MOD)) %= MOD;
return y;
}
ll mod_pow(ll x, ll p, ll MOD) {
ll a = 1;
while (p) {
if (p%2) a = a*x%MOD;
x = x*x%MOD;
p/=2;
}
return a;
}
// mod_inverse
ll mod_inverse(ll a, ll m) {
return mod_pow(a, MOD-2, m);
}
// long long 専用 行列式を求める関数
number det(matrix A, const ll MOD) {
const int n = A.size();
assert(n == (int)A[0].size());
ll ans = 1;
for (int i = 0; i < n; i++) {
int pivot = -1;
for (int j = i; j < n; j++) if (A[j][i]) {
pivot = j;
break;
}
if (pivot == -1) return 0;
if (i!=pivot) {
swap(A[i], A[pivot]);
ans *= -1;
}
ll inv = mod_inverse(A[i][i], MOD);
for (int j = i+1; j < n; j++) {
ll c = A[j][i]*inv % MOD;
for (int k = i; k < n; k++) {
A[j][k] = (A[j][k] - c*A[i][k])%MOD;
}
}
(ans *= A[i][i]) %= MOD;
}
if (ans < 0) ans += MOD;
return ans;
}