-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathMatrix multiplication.cpp
More file actions
78 lines (72 loc) · 1.73 KB
/
Matrix multiplication.cpp
File metadata and controls
78 lines (72 loc) · 1.73 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
const int mod = 1e9 + 7;
int L;
int64_t p2[64];
vector<vector<vector<int>>> powers;
vector<vector<int>> mul(const vector<vector<int>> &a, const vector<vector<int>> &b) {
int n = a.size(), m = b[0].size();
vector<vector<int>> sol(n, vector<int>(m));
for (int i = 0; i < n; ++i) {
for (int j = 0; j < m; ++j) {
for (int k = 0; k < (int)b.size(); ++k) {
sol[i][j] += (int64_t)a[i][k] * b[k][j] % mod;
if (sol[i][j] >= mod) {
sol[i][j] -= mod;
}
}
}
}
return sol;
}
vector<vector<int>> sqr(const vector<vector<int>> &M) {
vector<vector<int>> sol(L, vector<int>(L));
for (int i = 0; i < L; ++i) {
for (int j = 0; j < L; ++j) {
for (int k = 0; k < L; ++k) {
sol[i][j] += (int64_t)M[i][k] * M[k][j] % mod;
if (sol[i][j] >= mod) {
sol[i][j] -= mod;
}
}
}
}
return sol;
}
void powers_of_two() {
p2[0] = 1;
for (int i = 1; i < 64; ++i) {
p2[i] = p2[i - 1] << 1LL;
}
}
void precompute() {
vector<vector<int>> M(L, vector<int>(L));
for (int i = 0; i < L - 1; ++i) {
M[i][i + 1] = 1;
}
for (int i = 0; i < L; ++i) {
M[L - 1][i] = a[i];
}
powers.clear();
powers.push_back(M);
for (int i = 1; i < 64; ++i) {
powers.push_back(sqr(powers.back()));
}
}
vector<int> multiply(const vector<int> &v, const vector<vector<int>> &M) {
vector<int> sol(L);
for (int i = 0; i < L; ++i) {
for (int j = 0; j < L; ++j) {
sol[i] += (int64_t)v[j] * M[j][i] % mod;
if (sol[i] >= mod) {
sol[i] -= mod;
}
}
}
return sol;
}
void compute(vector<int> &v, const int64_t &expo) {
for (int bit = 0; p2[bit] <= expo; ++bit) {
if (expo & p2[bit]) {
v = multiply(v, powers[bit]);
}
}
}