快速幂/快速乘/矩阵快速幂模板

一、快速乘

ll mul(ll m,ll k,ll p)
{
    ll ans=0;
    while(k)
    {
        if(k&1)
        ans=(ans+m)%p;
        m=m*2%p;
        k>>=1;
    }
    return ans;
}

二、快速幂

ll qmi(ll m,ll k,ll p)
{
    ll res=1%p,t=m%p;
    while(k)
    {
        if(k&1)
            res=res*t%p;
        t=t*t%p;
        k>>=1;
    }
    return res;
}

三、矩阵快速幂

const int N=10;
int tmp[N][N];
void mul(int a[][N],int b[][N],int n)
{
    memset(tmp,0,sizeof tmp);
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        for(int k=0;k<n;k++)
        tmp[i][j]+=a[i][k]*b[k][j];
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        a[i][j]=tmp[i][j];
}
int res[N][N];
void qmi(int a[][N],int n)
{
    memset(res,0,sizeof res);
    for(int i=0;i<N;i++) res[i][i]=1;
    while(n)
    {
        if(n&1)
            mul(res,a,N);
        mul(a,a,N);
    }
}

题目背景
矩阵快速幂

题目描述
给定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 算法：矩阵快速幂
 

#include<stdio.h>
#include<iostream>
#include<cstring>
using namespace std;
#define maxn 1005
typedef long long ll;
const int p=1e9+7;
ll n,k;
typedef struct mat
{
    ll m[maxn][maxn];
}Matrix;
Matrix Mul(Matrix a,Matrix b)
{
    Matrix ans;
    memset(ans.m,0,sizeof ans.m);
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            for(int k=0;k<n;k++)
            {
                ans.m[i][j]+=(a.m[i][k]%p)*(b.m[k][j]%p)%p;
                ans.m[i][j]%=p;
            }
        }
    }
    return ans;
}
Matrix qsm(Matrix a,ll b)
{
    Matrix res;
    memset(res.m,0,sizeof res.m);
    for(int i=0;i<n;i++)
        res.m[i][i]=1;
    while(b)
    {
        if(b&1) res=Mul(res,a);
        a=Mul(a,a);
        b>>=1;
    }
    return res;
}
int main(){
    Matrix mat,ans;
    cin>>n>>k;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        scanf("%lld",&mat.m[i][j]);
    }
    ans=qsm(mat,k);
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        printf("%lld ",ans.m[i][j]);
        printf("\n");
    }
    return 0;
}

在洛谷上可以过，但codeblocks崩