矩阵快速幂 北邮机试 矩阵幂

矩阵快速幂 北邮机试 矩阵幂

#include<bits/stdc++.h>
using namespace std;
int a[10][10],b[10][10],c[10][10],d[10][10];
int n,kk;
int main()
{
	cin>>n>>kk;
	for(int i=0;i<n;i++)
	    for(int j=0;j<n;j++)
	        cin>>a[i][j];
    for(int i=0;i<n;i++) c[i][i]=1;//将c矩阵初始化为单位矩阵 
	while(kk>0)
	{
		if(kk&1)//注意位运算的运算顺序 
		{
			fill(d[0],d[0]+10*10,0);//******清空 
			for(int i=0;i<n;i++)//控制左行 
			    for(int k=0;k<n;k++)//控制右列 
	                for(int j=0;j<n;j++)//控制走步 
	                    d[i][k]+=a[i][j]*c[j][k];
	        for(int i=0;i<n;i++)
	            for(int j=0;j<n;j++)
	                c[i][j]=d[i][j];//c矩阵存储最后结果 
		}//******每移位一次就要自乘一次 
		fill(d[0],d[0]+10*10,0);//******清空 
			for(int i=0;i<n;i++)//控制左行 
			    for(int k=0;k<n;k++)//控制右列 
	                for(int j=0;j<n;j++)//控制走步 
	                    d[i][k]+=a[i][j]*a[j][k];
	        for(int i=0;i<n;i++)
	            for(int j=0;j<n;j++)
	                a[i][j]=d[i][j];//a矩阵存储乘数 
		kk>>=1;//k移位 
	}
	for(int i=0;i<n;i++)
	{
	    for(int j=0;j<n;j++) cout<<c[i][j]<<" ";
	    cout<<endl;
	}
	return 0;
}

矩阵快速幂 北邮机试 矩阵幂