矩阵乘法

  基础练习 矩阵乘法  

时间限制：1.0s   内存限制：512.0MB

   

问题描述

　　给定一个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<stdio.h>
#include<iostream>
using namespace std;
int main()
{
	int N,M;
	cin>>N;
	cin>>M;
	int a[40][40];//输入的原始数组的值
	long long b[40][40];//存储最终的结果
	for(int i=0;i<N;i++)
	{
	for(int j=0;j<N;j++)
	{
	cin>>a[i][j];
	b[i][j]=a[i][j];
	}
	}
	long long t[40][40];//存储中间结果的值
	int k,l;
	if(M==0)//特殊情况当要求矩阵的零次幂为单位矩阵
	{
		for(k=0;k<N;k++)
		{
			for(l=0;l<N;l++)
			{
				if(k==l)
				cout<<"1"<<" ";
				else
				cout<<"0"<<" ";
			}
			cout<<endl;
		}
	}
	else
	{
		while(--M)
		{
			for(int w=0;w<N;w++)
			{
				for(int e=0;e<N;e++)
				{
					int p=N;
					while(p)
					{
						t[w][e]+=b[w][p-1]*a[p-1][e];//矩阵的乘法公式
						p--;
					}
				}
			}
			for(k=0;k<N;k++)
			{
				for(l=0;l<N;l++)
				{
					b[k][l]=t[k][l];
					t[k][l]=0;
				}
			}	
		}
	for(k=0;k<N;k++)
	{
		for(l=0;l<N;l++)
		{
			cout<<b[k][l]<<" ";
		}
		cout<<endl;
	}
	}
	return 0;
}