【数论】矩阵快速幂入门

今天给大家讲解一下关于矩阵快速幂的知识
那么想要了解矩阵快速幂，首先你得有个矩阵

矩阵 说白了就是二维数组 但是矩阵乘法有自己的一套运算法则

如图 这是怎么运算的呢？
3=2*1+1*1
7=4*1+3*1
4=2*2+1*0
8=4*2+3*0
也就是说矩阵乘法的运算前提就是a行b列的矩阵乘b行c列的矩阵得到一个a行c列的矩阵

其次 你还要明白快速幂
快速幂比较常见的写法有俩种

1.二分 n^p=n^(p/2)*n^(p/2)
2.二进制 将p分为2^p1+2^p2+2^p3…………

比较常见的是第二种 虽然我觉得俩种算法代价都是logP级别的

好了 我们进入正题 什么是矩阵快速幂？
就是讲矩阵看做一个数 然后用快速幂来写

【问题描述】
求斐波那契数列的第n项modp的结果

【问题分析】
斐波那契数列通项公式
f(n+2)=f(n)+f(n+1)
所以将ans矩阵初始化为单位矩阵(等价于数字1)
将basic矩阵初始化如下图

这里说一下几个比较重要的事情

1.什么是矩阵加速优化
矩阵优化可以讲很多无法乘方的方程改变为可以使用快速幂的log级别时间代价

2.怎样构造初始矩阵
比如斐波那契数列要构造一个basic矩阵使得
[a,b] *basic=[b,a+b]
所以可得basic的初始矩阵

#include <cstdio>
#include <iostream>
using namespace std;
struct zk {  int matrix[2][2];  }ans, base;
int n,p;
zk multi(zk a,zk b)//矩阵快速幂
{
    zk temp;   int i,j,k;
    for(i=0;i<2;i++)
        for(j=0;j<2;j++)
        {
            temp.matrix[i][j]=0;
            for(k=0;k<2;k++)
                temp.matrix[i][j]=(temp.matrix[i][j]+a.matrix[i][k]*b.matrix[k][j])%p;
                //矩阵乘法原理
        }
    return temp;
}
void work(int k)
{
    base.matrix[0][0]=1;    base.matrix[0][1]=1;
    base.matrix[1][0]=1;    base.matrix[1][1]=0;
    //初始化basic矩阵 作为累乘的基本单位
    ans.matrix[0][0]=1; ans.matrix[1][1]=1;
    ans.matrix[0][1]=0; ans.matrix[1][0]=0;
    //初始化ans
    while(k)
    {
        if(k & 1)
            ans=multi(ans, base);
        base=multi(base, base);
        k>>=1;
    }
    printf("%d\n",ans.matrix[0][1]);
    return;
}
void read()
{
    int i,t;
    scanf("%d%d",&n,&p);
    for (i=1;i<=n;i++)
    {
        scanf("%d",&t);
        work(t);
    }
    return;
}
int main()
{
    freopen("seat.in","r",stdin);
    freopen("seat.out","w",stdout);
    read();
    return 0;
}