基础训练 矩阵乘法

矩阵乘法

#include<iostream>
#include<vector>
using namespace std;
int n, m;
vector<vector<int> > Mul(vector<vector<int> > t1, vector<vector<int> > t2){
	vector<vector<int> > temp(n, vector<int>(n, 0));
	for(int i=0; i<n; i++)
		for(int j=0; j<n; j++)
			for(int k=0; k<n; k++)
				temp[i][j]+=t1[i][k]*t2[k][j];
	return temp;	
} 
vector<vector<int> > QuickMul(vector<vector<int> > A, vector<vector<int> > ans, int m){
	while(m){
		if(m&1)
			ans=Mul(ans, A);
		A=Mul(A, A);
		m=m>>1;
	}
	return ans;
}
int main(){
	cin>>n>>m;
	vector<vector<int> > ans(n, vector<int>(n,0)), A(n, vector<int>(n,0));
	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++)
		for(int j=0; j<n; j++)
			if(i==j)
				ans[i][j]=1;
	ans=QuickMul(A, ans, m); 
	for(int i=0; i<n; i++){
		int flag=0;
		for(int j=0; j<n; j++)
			if(flag++==0) cout<<ans[i][j];
			else cout<<" "<<ans[i][j];
		cout<<endl; 
	}
	return 0; 
}