矩阵快速幂算法

来源:互联网 发布:windows gdi 图形编程 编辑:程序博客网 时间:2024/06/16 12:44

矩阵快速幂就是用来快速的求取矩阵的高次方的算法。

它将普通的时间复杂度O(n)降到了log(n)。

按照常理来求A^7的话,那就是A*A*A*A*A*A*A来得到去我们想要的答案,但是如果要是求A^100000000次方呢,是不是顿时感觉不想算了(想算也算不了)。

这时我们就该想到矩阵的快速幂了。

矩阵的快速幂主要还是运用到计算机的二进制处理喽。

现在给个例子:

计算A^7,首先我们把幂变成二进制,那么A^7 = A^(111)2,这样我们就可以根据代码来进行理解了。

首先我们知道n = 7,base保存的就是矩阵A, res开始时赋为单位矩阵(单位矩阵为除了对角线都为1,其他都为0)。

while(n)
   {
        if(n&1) ///求n是不是偶数  等价于n%2==1
            mulit(res, base);
        mulit(base, base);
        n>>=1;  (也可以写成n/=2)
   }

执行过程就是:

res=1;base=A; 

N=7 ;N%2=1;   res*=base; 所以res=A;  base*=base; 所以base=A^2

然后 N/=2;N=3; N%2=1; res*=base; 所以 res=A*A^2=A^3 ; 又base*=base; 所以base=A^4

然后N/=2;N=1;N%2=1;res*=base; 所以 res=A*A^2*A^4=A^7;又base*=base; 所以base=A^8

然后N/=2;N=0;算法结束

关键代码:

void mulit(int a[][2], int b[][2])
{
    int temp[2][2];
    for(int i = 0; i < 2; i++)
    {
        for(int j = 0; j < 2; j++)
        {
            temp[i][j] = 0;
            for(int k = 0; k < 2; k++)
            {
                temp[i][j] += (a[i][k] * b[k][j]) % 10000;
            }
        }
    }
    for(int i = 0; i < 2; i++)
    {
        for(int j = 0; j < 2; j++)
        {
            a[i][j] = temp[i][j] % 10000;
        }
    }

}
int solve(int n)
{
    int base[2][2] = {{1, 1},{1, 0}};
    int res[2][2] = {{1, 0},{0, 1}};
    while(n)
    {
        if(n&1) ///求n是不是偶数  等价于n%2==1
            mulit(res, base);   ///在mulit中进行了对res数组的赋值
        mulit(base, base);
        n>>=1;
    }
    return res[0][1];
}


这样是不是就好理解了呢。

当然要更好的理解还是少不了根据题来理解。

推荐一道题:

http://poj.org/problem?id=3070


求斐波纳挈的n次幂。

根据矩阵快速幂来求:

AC代码:

#include <iostream>#include <cstdio>#include <cstring>using namespace std;void mulit(int a[][2], int b[][2]){    int temp[2][2];    for(int i = 0; i < 2; i++)    {        for(int j = 0; j < 2; j++)        {            temp[i][j] = 0;            for(int k = 0; k < 2; k++)            {                temp[i][j] += (a[i][k] * b[k][j]) % 10000;            }        }    }    for(int i = 0; i < 2; i++)    {        for(int j = 0; j < 2; j++)        {            a[i][j] = temp[i][j] % 10000;        }    }}int solve(int n){    int base[2][2] = {{1, 1},{1, 0}};    int res[2][2] = {{1, 0},{0, 1}};    while(n)    {        if(n&1) ///求n是不是偶数  等价于n%2==1            mulit(res, base);        mulit(base, base);        n>>=1;    }    return res[0][1];}int main(){    int t;    while(~scanf("%d",&t))    {        if(t==-1)            break;        int ans = solve(t);        printf("%d\n",ans);    }    return 0;}

本人菜鸟,暂时就理解这么多,欢迎来提意见。


0 0
原创粉丝点击