[原博客存档] [模板] 矩阵快速幂

Luogu P3390 [模板]矩阵快速幂

题目背景

矩阵快速幂

题目描述

给定n*n的矩阵A，求A^k

输入输出格式

输入格式：
第一行，n,k

第2至n+1行，每行n个数，第i+1行第j个数表示矩阵第i行第j列的元素

输出格式：
输出A^k

共n行，每行n个数，第i行第j个数表示矩阵第i行第j列的元素，每个元素模10^9+7

输入输出样例

输入样例#1：

2 1
1 1
1 1

输出样例#1：

1 1
1 1

说明

n<=100, k<=10^12, |矩阵元素|<=1000 算法：矩阵快速幂

Solution:

给定一个n*m的矩阵A和m*k的矩阵B，A×B得到一个n*k的矩阵C。

怎么理解呢？矩阵C的某行某列就是A的某行乘以B的某列。

某行乘以某列就是行上的元素分别与列上的元素按顺序相乘最后相加的结果。

所以对于矩阵C，m根本不重要，n和k决定矩阵C的大小。

那么矩阵A和矩阵B为什么一定要求m相等呢？

原因是如果m不等的话某行某列相乘就会有某些元素不够乘。

什么是单位矩阵呢？使得任意矩阵A有S*A=A的矩阵S就叫单位矩阵。相当于乘法里的1。

单位矩阵的行数列数相等。

单位矩阵的构造就是左上开始向右下的45度斜线上的元素是1，其余是0。

我们使用结构体加上运算符重载来实现这一操作。

Codes:

#include<bits/stdc++.h>
using namespace std;
const int MOD=1e9+7;
typedef long long ll;

int n;
struct matrix{
    static const short N=110;
    int a[N][N];
    matrix(bool init=false){
        memset(a,0,sizeof(a));
        if(init){
            for(register int i=0;i<n;i++)
                a[i][i]=1;
        } // 初始化单位矩阵
    }
    int* operator[](int x){return a[x];}
    matrix operator *(matrix &b){  // 不能用const（重点）
        matrix c;
        for(register int i=0;i<n;i++) // 枚举行
        for(register int j=0;j<n;j++) // 枚举行与列的元素
        for(register int k=0;k<n;k++) // 枚举列
        c[i][k]=(c[i][k]+(1ll*a[i][j]*b[j][k])%MOD)%MOD;
        return c;
    }
};

int main(){
    ll k;
    scanf("%d%lld",&n,&k);
    matrix x,ans(true);
    for(int i=0;i<n;i++)
    for(int j=0;j<n;j++)
        scanf("%d",&x[i][j]);
    while(k){
        if(k&1ll)ans=ans*x;
        x=x*x;
        k>>=1ll;
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(j)putchar(' ');
            printf("%d",ans[i][j]);
        }
        putchar('
');
    }
    return 0;
}