矩阵快速幂

Posted 2021-01-10 yanchaoyi

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了矩阵快速幂相关的知识，希望对你有一定的参考价值。

一、前期铺垫

　在讲矩阵快速幂之前，我们先来看一下整数快速幂。求 X 的 N 次方。

　举个例子，在求 x^19时，我们可以拆分成 x^16、x^2 和 x的乘积。我们观察19的二进制数（10011），发现二进制第 i 位上的值为 1 ，在乘积中就要有 x 的 2^i 的一项。据此我们可以利用遍历二进制数的每一位快速求出 X^N。

　代码如下：

 1 ll QuickPow(ll x,ll n)
 2 {
 3     ll tmp = (ll)x;
 4     ll res = 1;
 5     for(ll i=0;(1<<i)<=n;i++)
 6     {
 7         if(n&(1<<i))
 8         {
 9             res=(res*tmp)%mod;
10         }
11         tmp=(tmp*tmp)%mod;
12     }
13     return res;
14 }

也可以写成下面这样：

 1 ll QuickPow(ll x,ll n)
 2 {
 3     ll tmp = x;
 4     ll res=1;
 5     while(n)
 6     {
 7         if(n&1)
 8             res=(res*tmp)%mod;
 9         tmp = (tmp*tmp)%mod;
10         n>>=1;
11     }
12     return res;
13 }

二、矩阵快速幂的实现过程

　　现在问题变成求解矩阵 A 的 N 次方，我们可以类比整数快速幂，写一个矩阵的结构体，用一个matmul函数来定义矩阵的乘法，具体实现过程与整数快速幂类似。

struct mat
{
    ll m[maxn][maxn];
}unit;
void init()
{
    for(int i=1;i<maxn;i++)
        unit.m[i][i]=1;
}
mat matmul(mat a,mat b)
{
    mat ans;
    ll tmp =0;
    for(int i=1;i<maxn;i++)
    {
        for(int j=1;j<maxn;j++)
        {
            tmp=0;
            for(int k=1;k<maxn;k++)
            {
                tmp=(tmp+a.m[i][k]*b.m[k][j])%mod;
            }
            ans.m[i][j]=tmp;
        }
    }
    return ans;
}
mat QuickPow(mat a,ll n)
{
    mat tmp = a;
    mat res=unit;
    while(n)
    {
        if(n&1)
            res=matmul(res,tmp);
        tmp = matmul(tmp,tmp);
        n>>=1;
    }
    return res;
}