Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <cstring>
- using namespace std;
- const int MOD = 1000000007;
- void matmul(int mat1[][2], int mat2[][2], int mat3[][2]) {
- int tmp[2][2] = {};
- for (int i = 0; i < 2; ++i) {
- for (int j = 0; j < 2; ++j) {
- for (int k = 0; k < 2; ++k) {
- tmp[i][j] = (tmp[i][j] + (long long)mat1[i][k] * mat2[k][j] % MOD + MOD) % MOD;
- }
- }
- }
- memcpy(mat3, tmp, sizeof(tmp));
- }
- class Chimney {
- public:
- int countWays(long long n) {
- int mat[2][2] = { { 56, -64 }, { 8, -8 } };
- int res[2][2] = { { 24, 0 }, { 4, 0 } };
- --n;
- for (; n; n /= 2) {
- if (n % 2) matmul(mat, res, res);
- matmul(mat, mat, mat);
- }
- return res[0][0];
- }
- };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement