NYOJ 299

(前言:这是一道关于矩阵快速幂的问题,介绍矩阵快速幂之前,首先看“快速幂”问题。 在前面的博客里有记录到快速幂取模算法,不过总体的思想总是和取模运算混淆在一起,而忽略了“快速幂”运算本身。计算ab本来就是一个可以加速的过程,“快速幂取模”运算只不过是“快速幂”算法的一个应用罢了。)

一、快速幂运算

这次我们关注快速幂本身:

我们知道离散化处理信息是计算机的热点,把连续数据存储为二进制离散数据是计算机的硬件要求。那么快速幂运算能否也运用这一思想呢,答案是肯定的。

容易发现:A19  =  (A16)*(A2)*(A1),而我们在计算A16的过程中可以通过A8*A8来得到,而A8同样可以通过A4*A4来得到,这样以来本来需要16次乘法运算得到的只需要log(16)=4次即可。

  A*A = A2

  A2*A2 = A4

  A4*A4 = A8

  A8*A8 =A16

再举例计算A148.我们知道14810=100101002,也就有A148=A128*A16*A4. 观察二进制形式的因子,容易发现这样的规律早就在二进制表示中显露出来了:100101002=100000002+000100002+000001002 =128+16+4。

核心代码表示为:  (第3行代码每进行一次,二进制数就少了最后面的一个1。二进制数有多少个1就第3行代码就执行多少次。)

 //在一路的A^x次方(x=1,2,4,8,16,...)的结果中,res通过“n的二进制位的值”来选取组成自己的“元素”
 while(n){
     )
         res=res*A;
     n>>=;
     A=A*A;
 }    

测试:

 #include <iostream>
 using namespace std;
 //快速计算 A^n
 int main(){
     int A,n;
     ;
     cin>>A>>n;

     //在一路的A^x次方(x=1,2,4,8,16,...)的结果中,res通过“n的二进制位的值”来选取组成自己的“元素”
     while(n){
         )
             res=res*A;
         n>>=;
         A=A*A;
     }
     cout<<res;
 } 

有输入的A^n

二、快速幂取模   ab%c

现在对于ab我们已经拿到尚方宝剑了,我们知道指数运算往往是很容易突破数据储存长度的(long long int 也才64位),所以一些方案处理中要求的只是ab%c而已。(反正这类问题客观存在,你就当我为它的存在杜撰了一个理由吧)

这时候可以运用取模运算的性质:(a*a)%c = (a%c)(a%c)%c

那这样取模运算就可以“镶嵌”在快速幂运算中了:   (if语句中“b%2==1”和上面的“n&1”的写法是一样的)

 ;
 a%=c;
 ){
     ==){
         num=(num*a)%c;
     }
     b>>;     //这一步将b->log2(b)
     a=(a*a)%c;
 }

三、矩阵快速幂

矩阵快速幂的思想就是跟数的快速幂一样,不过是把“数”换成“矩阵”罢了,在之前的数的乘法运算是用“*”直接运算,而这里需要单独写一个两矩阵相乘的函数以供调用。

这里展示一个示例程序:

 #include <cstdlib>
 #include <cstring>
 #include <cstdio>
 #include <iostream>
 using namespace std;

 int N;

 struct matrix{
        ][];
 }origin,res;

 matrix multiply(matrix x,matrix y){
        matrix temp;
        memset(temp.a,,sizeof(temp.a));
        ;i<;i++) {
                ;j<;j++) {
                        ;k<;k++) {
                                temp.a[i][j]+=x.a[i][k]*y.a[k][j];
                        }
                }
        }
        return temp;
 }

 void init(){
      printf("随机数组如下:\n");
      ;i<;i++){
              ;j<;j++){
                      origin.a[i][j]=rand()%;
                      printf("%8d",origin.a[i][j]);
              }
              printf("\n");
      }
      printf("\n");
      memset(res.a,,sizeof(res.a));
      res.a[][]=res.a[][]=res.a[][]=;                  //将res.a初始化为单位矩阵
 }

 void calc(int n){
      printf("%d次幂结果如下:\n",n);
      while(n) {
              )
                     res=multiply(res,origin);
              n>>=;
              origin=multiply(origin,origin);
      } 

      ;i<;i++) {
              ;j<;j++)
                      printf("%8d",res.a[i][j]);
              printf("\n");
      }
      printf("\n");
 }

 int main(){
     while(cin>>N) {
             init();
             calc(N);
     }
     ;
 }

矩阵 快速幂 演示

四、矩阵快速幂的应用

矩阵题目的难点在于构造矩阵,一般用于有能够推出递推式的题目,推出递推式之后,发现递推O(n)的时间复杂度比较大,那么我们可以构造一个矩阵,然后用矩阵快速幂降低到log(n)的时间复杂度

在NYOJ 299中

Fk = A + A2 + A3 + … + Ak

F(k-1) = A + A2 + A3 + … + Ak-1

故 Fk = A + A*F(k-1)

那么可以构造矩阵:

(Fk ,1 ) =  (Fk-1 ,1) * (A,0; A,1) = (F1 , 1) * (A,0;A,1)K-1

这样就可以运用矩阵快速幂了。

参考:

2. http://www.cnblogs.com/yan-boy/archive/2012/11/29/2795294.html

3. http://blog.csdn.net/chenguolinblog/article/details/10309423

4. http://blog.csdn.net/y990041769/article/details/39694583

5. http://blog.csdn.net/y990041769/article/details/39716951

6. http://www.matrix67.com/blog/archives/276

上一篇:NYOJ 998


下一篇:Manacher算法 , 实例 详解 . NYOJ 最长回文