东华OJ挑战-9 矩阵乘法

作者: Turbo时间限制: 1S章节: 基本练习（循环）

问题描述

给定一个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<bits/stdc++.h>
 
using namespace std;
 
int a[50][50],c[50][50],b[50][50];
 
int main()
{
	//freopen("in.txt","r",stdin);
	int n,m,i,j,k,x;
	scanf("%d %d",&n,&m);
	for (i=0;i<n;i++)
		for (j=0;j<n;j++)
			scanf("%d",&a[i][j]);
	memset(c,0,sizeof(c));
	for (i=0;i<n;i++)
		c[i][i] = 1;//单位矩阵 
	while(m--)
	{
		for (k=0;k<n;k++)
		{
			for (i=0;i<n;i++)
			{
				x = 0;
				for(j=0;j<n;j++)
					x += c[i][j] * a[j][k];
				b[i][k] = x;
			}
		}
		for (i=0;i<n;i++)
			for (j=0;j<n;j++)
				c[i][j] = b[i][j];
	}
	for (i=0;i<n;i++)
	{
		for (j=0;j<n-1;j++)
			printf("%d ",c[i][j]);
		printf("%d\n",c[i][n-1]);
	
	}
	return 0;
}

运行状态：AC