【蓝桥杯】基础训练 矩阵乘法（方阵的幂）

memset(t,0,sizeof(t));

给定一个N阶矩阵A，输出A的M次幂（M是非负整数）

例如：

A  =

1  2

3  4

A的2次幂

7  10

15  22

输入

第一行是一个正整数N、M（1< =N< =30,  0< =M< =5），表示矩阵A的阶数和要求的幂数 

接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值 

输出

输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开

样例输入

2 2 
1 2 
3 4 

样例输出

7 10
15 22

---

---

线代知识：方阵相乘，第一行第一个数，等于方阵第一行与第一列相乘之和。第一行第二个数等于方阵第一行与第二列相乘之和。

即 t[i][j]=r[i][k]*a[k][j]; t[i][j]相当于sum=0；（相加用） r[i][j]相当于1；(相乘用)

for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				for(int k=0;k<n;k++){
					t[i][j]+=r[i][k]*a[k][j];
					}}}

memset(t,0,sizeof(t));  初始化函数（初始化t）

代码：

#include<bits/stdc++.h>
using namespace std;
int main()
{    int a[100][100],t[100][100],r[100][100];
	int n,m;
	cin>>n>>m;
	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++) 
 		r[i][i]=1;
	while(m--)
	{
		memset(t,0,sizeof(t));
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				for(int k=0;k<n;k++){
					t[i][j]+=r[i][k]*a[k][j];
					}}}
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
		 		r[i][j]=t[i][j];}}
	}
	for( int i=0;i<n;i++){
		for(int j=0;j<n;j++)
			cout<<r[i][j]<<" ";
			cout<<endl;
	}              
	return 0;
}