矩阵快速幂概述

推荐一波博客:传送门!!!

矩阵乘法可以用于某些dp当中进行优化,或者优化一些数论算法。这里就只讲解最简单的模板写法。以后再说实际应用。或者可以在上面博客上看一看。

首先普通快速幂的写法是很容易想出来的,就是一个二分,应用分治的思想。如果求出了 k2 k 2 ,那么可以很快求 k4 k 4 。看下代码:

#include<iostream>  
#include<cstdio>  
#include<cstring>  
#include<algorithm>  
using namespace std;  
long long c;  

long long quick(long long a,long long b)  
{  
    if(b==1)return a;  
    if((b^1)==1)  return (quick((a*a)%c,b>>1)*a)%c;  
    if((b^2)==0)  return (quick((a*a)%c,b>>1));  
}  

int main()  
{  
    long long a,b;  
    scanf("%lld%lld%lld",&a,&b,&c);  
    printf("%lld^%lld mod %lld=%lld",a,b,c, (quick(a,b))%c);  
}  

唯一需要注意的就是取模。

然后我就傻逼了,一开始还以为矩阵快速幂有什么式子,结果也那么暴力。当然矩阵乘法能使用快速幂得益于它的一个性质:乘法结合律。(但是矩阵乘法没有交换律)。所以我们只需要用结构体封装一个矩阵乘法就行,然后直接套快速幂的板子。

比较简单的矩乘写法:

for(int i=1;i<=n;i++)  
  for(int j=1;j<=n;j++)  
    for(int k=1;k<=n;k++)  
      C.m[i][j]=(C.m[i][j]+A.m[i][k]*B.m[k][j])%mod;  

似乎还有一种写法(把别人博客的搬过来):

for(int i=1;i<=n;i++)    
  for(int k=1;k<=n;k++)    
    for(int j=1;j<=n;j++)    
      C.m[i][j]=(C.m[i][j]+A.m[i][k]*B.m[k][j])%mod;   

就是把循环交换一下,在第二重循环中可以判断A.m[i][k]==0,continue的话,据说对于0多的矩阵有奇效。

但是有一个问题,就是矩阵快速幂可能不那么支持递归,浪费空间而且windows还tm要爆栈。

我们研究一种非递归写法:

long long quick(long long b,long long x)  
{  
    int c=1;  
    while(x>0)  
    {  
        if((x&1)==1)  
          c=c*b%mod;  
        b=b*b%mod;  
        x=x>>1;  
    }  
    return c;  
}  

用二进制想一想:求 b1b2,b4,b8 b 1 , b 2 , b 4 , b 8 ……

然后把次数拆成二进制,位置上为1就乘到答案里。

其实也挺好理解的,还是建议写非递归吧,似乎适用面要广一些。

然后我们会快速幂,会矩乘,然后我们就会矩阵快速幂了(把俩板套在一起):

#include<iostream>  
#include<cstdio>  
#include<cstring>  
using namespace std;  
long long n,k;  
long long const mod=1000000007;  
struct lxy{  
    long long m[105][105];  
    lxy operator * (const lxy &q)  
    {  
        lxy C;  
        for(int i=1;i<=n;i++)  
          for(int j=1;j<=n;j++)  
            C.m[i][j]=0;  
        for(int i=1;i<=n;i++)  
          for(int j=1;j<=n;j++)  
            for(int k=1;k<=n;k++)  
              C.m[i][j]=(C.m[i][j]+m[i][k]*q.m[k][j])%mod;  
        return C;  
    }  
}A;  

lxy quick(lxy B,long long x)  
{  
    lxy C;  
    for(int i=1;i<=n;i++)  
      for(int j=1;j<=n;j++)  
      {  
        if(i==j) C.m[i][j]=1;  
        else C.m[i][j]=0;  
      }  
    while(x>0)  
    {  
        if((x&1)==1)  
          C=C*B;  
        B=B*B;  
        x=x>>1;  
    }  
    return C;  
}  

int main()  
{  
    scanf("%lld%lld",&n,&k);  
    for(int i=1;i<=n;i++)  
      for(int j=1;j<=n;j++)  
        scanf("%lld",&A.m[i][j]);  
    A=quick(A,k);  
    for(int i=1;i<=n;i++)  
    {  
      for(int j=1;j<=n;j++)  
        printf("%lld ",A.m[i][j]%mod);  
      printf("\n");  
    }  
}  

可以先看一看超大数斐波那契数列!!!:Fibonacci

然后分类在dp还是在数论里?

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值