当前位置:   article > 正文

高阶矩阵快速幂_矩阵高阶幂怎么求c语言

矩阵高阶幂怎么求c语言

前言

所谓的高阶矩阵快速幂就是在快速幂的基础上,结合矩阵运算计算矩阵的高次方的。它可以将朴素的o(n)的时间复杂度降为o(logn)。

    我们先对原理进行简单的介绍,对于快速幂来说,如果要求x^n,一般我们会将幂n 表示成 n = 2^k,那么x^n = ((x^2)^2)......,对于每个数我们只需要做k次平方运算就行了,因此我们只需要将n表示成2的幂次和。
 n = 2^k1 + 2^k2 + 2^k3 + ......,那么n就可以表示成x ^ n = x^(2^k1) * x^(2^k2) * x^(2^k3)......,依次求幂就行,这里给出快速幂的算法:

  1. typedef long long ll;
  2. ll mod_pow(ll x, ll n, ll mod)
  3. {
  4. ll res = 1;
  5. while(n > 0){
  6. if(n & 1)
  7. res = res * x %mod;
  8. //如果二进制的最低位为1,就要乘上x^(2^i)
  9. x = x * x % mod;
  10. n >>= 1;
  11. }
  12. return res;
  13. }
有了上面的知识,我们如何快速求的矩阵的N次幂呢?

矩阵快速幂

我们可以将矩阵的幂进行离散化,进行二进制表示例如求A^156 ,我们可以将156(10) = 10011100(2),也就是A^(156) = (A^4)*(A^8)*(A^16)*(A^128)
就可以将代码改成
  1. while(N)
  2. {
  3. if(N&1)
  4. res = res*A;
  5. n >>= 1;
  6. A = A*A;
  7. }

例子

斐波那契数列是由下面的递推定义的数列

F0 = 0
F1 = 1
Fn+2 = Fn+1 + Fn

求这个数列第n项的值对10^4取余后的结果

数据范围:0 <= n <= 10^16

分析:通过逐项求解这个递推式,我们可以在O(n)的时间内求解,但是效率太低。当然我们也可以求出通项公式:

来求解,但是因为其中含有无理数,所以我们不能进行求模运算。在这种情况下,我们就要使用矩阵快速幂了
首先,我们需要将递推公式表示成矩阵形式

记这个矩阵为A,那么



只要求出A^n的值,我们就可以求出Fn了,就可以在O(logn)的时间内求出第n项的值

标答如下:
  1. #include <cstdio>
  2. #include <cstring>
  3. #include <cstdlib>
  4. #include <vector>
  5. using namespace std;
  6. typedef vector<int> vec;
  7. typedef vector<vec> mat;
  8. typedef long long ll;
  9. const int M = 10000;
  10. //计算两个矩阵的积
  11. mat mul(mat &A, mat &B)
  12. {
  13. mat C(A.size(), vec(B[0].size()));
  14. for(int i=0;i< A.size(); i++)
  15. {
  16. for(int k=0;k<B.size();k++)
  17. {
  18. for(int j=0;j<B[0].size();j++)
  19. {
  20. C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % M;
  21. }
  22. }
  23. }
  24. return C;
  25. }
  26. //计算A^n
  27. mat pow(mat A, ll n)
  28. {
  29. mat B(A.size(), vec(A.size()));
  30. for(int i=0;i<A.size();i++)
  31. {
  32. B[i][i] = 1;
  33. }
  34. while(n > 0)
  35. {
  36. if(n&1)
  37. B = mul(B, A);
  38. A = mul(A, A);
  39. n >>= 1;
  40. }
  41. return B;
  42. }
  43. ll n;
  44. int main()
  45. {
  46. scanf("%lld",&n);
  47. mat A(2, vec(2));
  48. A[0][0] = 1;
  49. A[0][1] = 1;
  50. A[1][0] = 1;
  51. A[1][1] = 0;
  52. A = pow(A, n);
  53. printf("%d\n", A[1][0]);
  54. return 0;
  55. }

拓展

最后,我想谈一下自己关于二进制值在计算机中的用途,一方面自己以前在刷题过程中经常遇到使用二进制来降低时间复杂度的方法,另一方自己学了离散数学有感。

大家都知道计算机所能处理的信息是离散信息,就是以0,1作为信号进行处理。由此我们就可以知道二进制在计算中有着举足轻重的地位,他能将模拟型号转化成数字信号,将原来连续的模型,用一个离散的算法模型进行解决,在算法中就有许多用到离散化,比如

1.多重背包问题
2.线段树区间的离散化
3.树状数组
4.状态压缩DP
5.搜索中状态标记
.......



((x^2)^2)......,对于每个数我们只需要做k次平方运算就行了,因此我们只需要将n表示成2的幂次和 里给出快速幂的算
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/941926
推荐阅读
相关标签
  

闽ICP备14008679号