P3390 【模板】矩阵快速幂

http://www.cnblogs.com/yan-boy/archive/2012/11/29/2795294.html 参考博客！；

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

没看数据类型，提交两次不对， 后来更正， ，，，尴尬，总体来说知识点+1；
题解：直接套用模板

#include<iostream>
#include<string.h>
using namespace std;
const int mod=1e9+7;
long long n,m;
struct act{
    long long  a[105][105];
}ans,base;
void init(){//初始化
    memset(ans.a,0,sizeof(ans.a));
    for(int i=0;i<m;i++)for(int j=0;j<m;j++){
            cin>>base.a[i][j];
            ans.a[i][j]=i==j?1:0;  //单位矩阵对角线为1 其余为0
        }
}
act multiply(act x,act y){
    act temp;
    for(int i=0;i<m;i++)for(int j=0;j<m;j++){
            temp.a[i][j]=0;
            for(int k=0;k<m;k++)  //k是列数
                temp.a[i][j]=(temp.a[i][j]+(x.a[i][k]*y.a[k][j])%mod)%mod;
        }
   return temp;//返回的是结构体类型
}
void calc(long long n){
    while(n){
        if(n&1) ans=multiply(ans,base);
        base=multiply(base,base);
        n>>=1;
    }
    for(int i=0;i<m;i++){for(int j=0;j<m;j++)   cout<<ans.a[i][j]<<" ";
        cout<<endl;
    }
    cout<<endl;
}
int main(){
    cin>>m>>n;init();calc(n);
    return 0;
}

//以这个为学习制作 工具的模板了  ，以后做自己可以用的工具


#include<iostream>
#include<cstdio>
#define For(i,a,b) for(register int i=a;i<=b;++i)
#define LL long long
const int mod=1e9+7;
using namespace std;
LL read(){
    char x=getchar(); LL u=0,fg=0;
    while(!isdigit(x)){if(x=='-') fg=1; x=getchar();}
    while(isdigit(x)) u=(u<<3)+(u<<1)+(x^'0'), x=getchar();
    return !fg?u:-u;
}
struct Matrix{
    LL n,k,a[110][110];
    void work(bool type){                    //work(0)是读入,work(1)是输出，所以代码短啊~
        if(!type){ n=read(); k=read();}
        For(i,1,n){
            if(i >1 && type)printf("\n");
            For(j,1,n) if(!type) a[i][j]=read(); else printf("%d ",a[i][j]);
        }
    }
    Matrix operator *(const Matrix &b) const{            //矩阵乘法
        Matrix res=b;
        For(i,1,n) For(j,1,n){
            res.a[i][j]=0;                        //res赋初值
            For(k,1,n) res.a[i][j]=(res.a[i][j]+a[i][k]*b.a[k][j]%mod)%mod;    //记得取模
        }
        return res;
    }
    friend Matrix operator ^(Matrix x, LL num){  //矩阵快速幂
        Matrix res=x;        //res=x，其实只要把x的n值赋给res
        For(i,1,res.n) For(j,1,res.n)
            if(i==j) res.a[i][j]=1;                //单位矩阵的主对角线为1
            else res.a[i][j]=0;
        while(num){
            if(num&1) res=res*x;
            x=x*x;
            num>>=1;
        }
        return res;
    }
}x;
int main(){
    x.work(0); x=x^x.k; x.work(1); return 0;
}