记录编号 437938 评测结果 AAAAAAAAAA
题目名称 斐波那契平方和 最终得分 100
用户昵称 GravatarHeHe 是否通过 通过
代码语言 C++ 运行时间 0.003 s
提交时间 2017-08-14 21:35:57 内存使用 0.32 MiB
显示代码纯文本
  1. #include <iostream>
  2. #include <cstdio>
  3. #include <cstring>
  4. using namespace std;
  5.  
  6. #define MOD (1000000007)
  7. typedef long long LL;
  8.  
  9. inline void Pow(LL N);
  10.  
  11. LL res[2][2] = {{1, 0},
  12. {0, 1}};
  13. LL fib[2] = {0, 1};
  14. LL tmp[2][2];
  15. LL q[2][2] = {{0, 1},
  16. {1, 1}};
  17. LL N;
  18.  
  19. int main() {
  20. #ifndef LOCAL
  21. freopen("fibsqr.in", "r", stdin);
  22. freopen("fibsqr.out", "w", stdout);
  23. #endif
  24. scanf("%lld", &N);
  25. Pow(N);
  26. memset(tmp, 0x00, sizeof(tmp));
  27. for(int i = 0; i < 2; ++i)
  28. for(int j = 0; j < 2; ++j)
  29. (tmp[0][i] += fib[j] * res[j][i]) %= MOD;
  30. printf("%lld", tmp[0][0] * tmp[0][1] % MOD);
  31. return 0;
  32. }
  33.  
  34. inline void Pow(LL N) {
  35. while(N) {
  36. if(N & 1) {
  37. memset(tmp, 0x00, sizeof(tmp));
  38. for(char i = 0; i < 2; ++i)
  39. for(char j = 0; j < 2; ++j)
  40. for(char k = 0; k < 2; ++k)
  41. (tmp[i][j] += res[i][k] * q[k][j]) %= MOD;
  42. for(char i = 0; i < 2; ++i)
  43. for(char j = 0; j < 2; ++j)
  44. res[i][j] = tmp[i][j];
  45. }
  46. memset(tmp, 0x00, sizeof(tmp));
  47. for(char i = 0; i < 2; ++i)
  48. for(char j = 0; j < 2; ++j)
  49. for(char k = 0; k < 2; ++k)
  50. (tmp[i][j] += q[i][k] * q[k][j]) %= MOD;
  51. for(char i = 0; i < 2; ++i)
  52. for(char j = 0; j < 2; ++j)
  53. q[i][j] = tmp[i][j];
  54. N >>= 1;
  55. }
  56. return ;
  57. }