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;
}