-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy path3700.cpp
More file actions
80 lines (76 loc) · 2.32 KB
/
Copy path3700.cpp
File metadata and controls
80 lines (76 loc) · 2.32 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
using ll = long long;
ll MOD = 1e9 + 7;
class Solution {
public:
vector<vector<ll>> matMul(const vector<vector<ll>>& A, const vector<vector<ll>>& B) {
int n = A.size();
int m = B[0].size();
int K = B.size(); // A: n x K, B: K x m
vector<vector<ll>> C(n, vector<ll>(m, 0));
for (int i = 0; i < n; ++i) {
for (int k = 0; k < K; ++k) {
ll aik = A[i][k];
if (aik == 0) continue;
for (int j = 0; j < m; ++j) {
C[i][j] += (aik * B[k][j]) % MOD;
if (C[i][j] >= MOD) C[i][j] -= MOD;
}
}
}
return C;
}
vector<vector<ll>> matMulSquare(const vector<vector<ll>>& A, const vector<vector<ll>>& B) {
return matMul(A, B);
}
vector<vector<ll>> matPow(vector<vector<ll>> base, long long e) {
int K = base.size();
// initialize res = identity
vector<vector<ll>> res(K, vector<ll>(K, 0));
for (int i = 0; i < K; ++i) res[i][i] = 1;
while (e > 0) {
if (e & 1) res = matMul(res, base);
base = matMul(base, base);
e >>= 1;
}
return res;
}
vector<ll> matVecMul(const vector<vector<ll>>& M, const vector<ll>& v) {
int n = M.size();
int m = v.size();
vector<ll> r(n, 0);
for (int i = 0; i < n; ++i) {
ll sum = 0;
for (int j = 0; j < m; ++j) {
if (M[i][j] == 0) continue;
sum += (M[i][j] * v[j]) % MOD;
if (sum >= MOD) sum -= MOD;
}
r[i] = sum % MOD;
}
return r;
}
int zigZagArrays(int n, int l, int r) {
int m = r - l + 1;
int k = 2 * m;
vector<vector<ll>> T(k, vector<ll>(k, 0));
for (int i = 0; i < m; ++i) {
for (int j = 0; j < i; ++j) {
T[i][m + j] = 1;
}
}
for (int i = 0; i < m; ++i) {
for (int j = i + 1; j < m; ++j) {
T[i + m][j] = 1;
}
}
vector<vector<ll>> Tn = matPow(T, n - 1);
vector<ll> v(k, 1);
vector<ll> vn = matVecMul(Tn, v);
ll res = 0;
for (auto x : vn) {
res += x;
res %= MOD;
}
return res;
}
};