蓝桥杯基础练习（矩阵乘法）

蓝桥杯基础练习（矩阵乘法）

资源限制
时间限制：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 <string.h> 
int r[50][50];
int s[50][50];
int t[50][50];

int main()
{
	int n, m;
	scanf("%d%d", &n, &m);
	int i, j, k;

	//赋值
	for (i = 1; i <= n; i++)
	{
		for (j = 1; j <= n; j++)
		{
			scanf("%d", &s[i][j]);
		}
	}
	for (i = 1; i <= n; i++)//初始化r
		r[i][i] = 1;

	while (m--)//m--先执行后减一
	{
		memset(t, 0, sizeof(t));//初始化数组t的元素全部为0，必须有这一步才能进行后面的求和操作
		for (i = 1; i <= n; i++)
		{
			for (j = 1; j <= n; j++)
			{
				for (k = 1; k <= n; k++)
				{
					t[i][j] += r[i][k] * s[k][j];//数组乘法规律
				}
			}
		}
		for (i = 1; i <= n; i++)//将数组t拷贝到数组r
		{
			for (j = 1; j <= n; j++)
				r[i][j] = t[i][j];
		}
	}
	//输出
	for (i = 1; i <= n; i++)
	{
		for (j = 1; j <= n; j++)
		{
			printf("%d ", r[i][j]);
		}
		printf("\n");
	}
	return 0;
}

memset函数介绍