矩阵快速幂(51nod)

1113 矩阵快速幂
基准时间限制:3 秒 空间限制:131072 KB 分值: 40 难度:4级算法题
给出一个N * N的矩阵,其中的元素均为正整数。求这个矩阵的M次方。由于M次方的计算结果太大,只需要输出每个元素Mod (10^9 + 7)的结果。

Input
第1行:2个数N和M,中间用空格分隔。N为矩阵的大小,M为M次方。(2 <= N <= 100, 1 <= M <= 10^9)
第2 - N + 1行:每行N个数,对应N * N矩阵中的1行。(0 <= N[i] <= 10^9)
Output
共N行,每行N个数,对应M次方Mod (10^9 + 7)的结果。
Input示例
2 3
1 1
1 1
Output示例
4 4
4 4

/*
代码是看的大佬的博客写出来的,就先当模板来理解吧
*/
#include
#include
#include
#include
using namespace std;
typedef long long ll;
ll mod=1e9+7;// 相当于宏定义 mod
int n;
typedef struct node//定义结构体,为接下来的代码设置矩阵方便许多
{
    ll a[101][101];
}matrix;
matrix mat_mul(matrix p,matrix q)//返回的是结构体类型的矩阵,所以定义成结构体类型的被调函数
{
    matrix res;//结果矩阵
    int i,j,k;
    ll x;
    memset(res.a,0,sizeof(res.a));
    for(i=0;i<=n-1;i++){       //矩阵乘法
        for(j=0;j<=n-1;j++){
            x=0;
            for(k=0;k<=n-1;k++){
                x+=(p.a[i][k]*q.a[k][j]);
                x%=mod; //先求完 和 之后在 mod
            }
            res.a[i][j]=x%mod; 防止数据超出范围,所以在 mod 一次
        }
    }
    return res; //返回矩阵
}
matrix quick_pow(matrix mat,ll m)//矩阵快速幂的模板
{
    matrix unit;
    int i;
    memset(unit.a,0,sizeof(unit.a));
    for(i=0;i<=n-1;i++)
        unit.a[i][i]=1;//设置单位矩阵,主对角线都是 1,其余是 0
    while(m){
        if(m&1)
            unit=mat_mul(unit,mat);//相当于 ans=mat_mul(E,mat);
        mat=mat_mul(mat,mat);      //      mat=mat_mul(mat,mat); 顺序不能变
        m>>=1;
    }
    return unit;//返回结果
}
int main()
{
    int i,j;
    ll m;
    scanf("%d %lld",&n,&m);
    matrix s,ans;
    for(i=0;i<=n-1;i++)
        for(j=0;j<=n-1;j++)
            scanf("%lld",&s.a[i][j]);
    ans=quick_pow(s,m);
    for(i=0;i<=n-1;i++){
        for(j=0;j<=n-1;j++){
            if(j

你可能感兴趣的:(ACM,----,算法,ACM,----,数据结构,ACM,----,知识点)