（蓝桥杯）矩阵相乘

矩阵相乘

问题描述

给定一个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

代码

#include <iostream>
using namespace std;
int a[30][30],b[30][30],c[30][30];
int main()
{
	int N,M;//N阶矩阵,M次幂 
	cin>>N>>M;
	for(int i=0;i<N;i++)
	for(int j=0;j<N;j++)
		{
			cin>>a[i][j];
			b[i][j]=a[i][j];
		}
		//尤其要注意，0次幂的情况要单独考虑 
	if(M==0)
	{
		for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
		{
          if(i==j)
          c[i][j]=1;
          else
          c[i][j]=0;	
		}
		   for(int i=0; i<N; i++)
    {
        for(int j=0; j<N; j++)
            cout<<c[i][j]<<" ";
        cout<<endl;
    }
		return 0;
	}
		
	for(int i=0;i<M-1;i++)
	{
		
		for(int j=0;j<N;j++)
		for(int k=0;k<N;k++)
		for(int l=0;l<N;l++)
		{
			c[j][k]+=a[j][l]*b[l][k];
		}
		
		for(int j=0;j<N;j++)
	    for(int k=0;k<N;k++)
	    {
	    	b[j][k]=c[j][k];
	    	c[j][k]=0;
		}
    }
    for(int i=0; i<N; i++)
    {
        for(int j=0; j<N; j++)
            printf("%d ",b[i][j]);
        printf("\n");
    }
	return 0;
 } 

总结

这题注意点有两个，一个是矩阵相乘时注意相乘的元素下标不要弄错，还有一个是注意0次幂的情况要单独考虑，自己做的时候也因为没考虑这一点，导致一直没有通过。