Advertisement
Guest User

Untitled

a guest
May 24th, 2017
59
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.74 KB | None | 0 0
  1. #include <cstring>
  2. using namespace std;
  3.  
  4. const int MOD = 1000000007;
  5.  
  6. void matmul(int mat1[][2], int mat2[][2], int mat3[][2]) {
  7. int tmp[2][2] = {};
  8. for (int i = 0; i < 2; ++i) {
  9. for (int j = 0; j < 2; ++j) {
  10. for (int k = 0; k < 2; ++k) {
  11. tmp[i][j] = (tmp[i][j] + (long long)mat1[i][k] * mat2[k][j] % MOD + MOD) % MOD;
  12. }
  13. }
  14. }
  15. memcpy(mat3, tmp, sizeof(tmp));
  16. }
  17.  
  18. class Chimney {
  19. public:
  20. int countWays(long long n) {
  21. int mat[2][2] = { { 56, -64 }, { 8, -8 } };
  22. int res[2][2] = { { 24, 0 }, { 4, 0 } };
  23. --n;
  24. for (; n; n /= 2) {
  25. if (n % 2) matmul(mat, res, res);
  26. matmul(mat, mat, mat);
  27. }
  28. return res[0][0];
  29. }
  30. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement