代码拉取完成,页面将自动刷新
// Time: O(logn)
// Space: O(logn)
class Solution {
public:
int numTilings(int N) {
vector<vector<int>> T{{1, 0, 0, 1}, // #(|) = #(|) + #(=)
{1, 0, 1, 0}, // #(「) = #(|) + #(L)
{1, 1, 0, 0}, // #(L) = #(|) + #(「)
{1, 1, 1, 0}}; // #(=) = #(|) + #(「) + #(L)
return matrixExpo(T, N)[0][0]; // T^N * [1, 0, 0, 0]
}
private:
vector<vector<int>> matrixExpo(const vector<vector<int>>& A, int pow) {
vector<vector<int>> result(A.size(), vector<int>(A.size()));
vector<vector<int>> A_exp(A);
for (int i = 0; i < A.size(); ++i) {
result[i][i] = 1;
}
while (pow) {
if (pow % 2 == 1) {
result = matrixMult(result, A_exp);
}
A_exp = matrixMult(A_exp, A_exp);
pow /= 2;
}
return result;
}
vector<vector<int>> matrixMult(const vector<vector<int>>& A, const vector<vector<int>>& B) {
vector<vector<int>> result(A.size(), vector<int>(A.size()));
for (int i = 0; i < A.size(); ++i) {
for (int j = 0; j < B[0].size(); ++j) {
int64_t entry = 0;
for (int k = 0; k < B.size(); ++k) {
entry = (static_cast<int64_t>(A[i][k]) * B[k][j] % M + entry) % M;
}
result[i][j] = static_cast<int>(entry);
}
}
return result;
}
const int M = 1e9 + 7;
};
// Time: O(n)
// Space: O(1)
class Solution2 {
public:
int numTilings(int N) {
// Prove:
// dp[n] = dp[n-1](|) + dp[n-2](=) + 2*(dp[n-3](「」) + ... + d[0](「 = ... = 」))
// = dp[n-1] + dp[n-2] + dp[n-3] + dp[n-3] + 2*(dp[n-4] + ... + d[0])
// = dp[n-1] + dp[n-3] + (dp[n-2] + dp[n-3] + 2*(dp[n-4] + ... + d[0])
// = dp[n-1] + dp[n-3] + dp[n-1]
// = 2*dp[n-1] + dp[n-3]
const int M = 1e9 + 7;
vector<int> dp(3);
dp[0] = 1, dp[1] = 1, dp[2] = 2;
for (int i = 3; i <= N; ++i) {
dp[i % 3] = (2 * dp[(i - 1) % 3] % M + dp[(i - 3) % 3]) % M;
}
return dp[N % 3];
}
};
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。