蓝桥杯 试题 基础练习 矩阵乘法 c语言蛮力法

快蓝桥杯了，好久没刷题，连这个都不会了，还得看网上的资料，这几天写博客记录每天刷的算法题，奥利给！

标题

题目
资源限制
时间限制：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

思路

n阶包括0和1还有非0非1三种，需要分类讨论（压根忘了矩阵乘法的规则了…这个坑一直在踩…），0次幂为单位矩阵，1次为本身，多次就是矩阵乘法。矩阵乘法来三个for就ok。

代码

#include<stdio.h>
int main()
{
	int n, m, t=1;//t为计数器（已经乘了多少次） 
	scanf("%d%d",&n,&m);
	//arr1：输入的矩阵  arr2：临时存储的矩阵  res：存结果的矩阵 
	int arr1[n][n], arr2[n][n], res[n][n];
	//输入 
	int i,j;
	int x;
	for(i=0;i<n;i++){
		for(j=0;j<n;j++){
			scanf("%d",&arr1[i][j]);
			arr2[i][j] = arr1[i][j];
		}
	}
	//0次幂
	if(m==0){
		for(i=0;i<n;i++){
			for(j=0;j<n;j++){
				res[i][j] = 0;
				if(i==j)
					res[i][j]++;
			}
		}
	}
	//1次幂
	else if(m==1) {
		for(i=0;i<n;i++){
			for(j=0;j<n;j++){
				res[i][j] = arr1[i][j];
			}
		}
	}
	//非0非1
	else{
		while(t<m){
			for(i=0;i<n;i++){
				for(j=0;j<n;j++){
					res[i][j] = 0;
					for(x = 0;x<n;x++){
						res[i][j] += arr1[i][x]*arr2[x][j];//矩阵乘法的定义来算
					}
				}
			}
			for(i=0;i<n;i++){
			    for(j=0;j<n;j++){
    				arr2[i][j]=res[i][j];
				}
			}
			t++;
		}
	} 
	//格式化输出 
	for(i=0;i<n;i++){
		for(j=0;j<n;j++){
			if(0==j)
				printf("%d",res[i][j]);
			else
				printf(" %d",res[i][j]);
		}
		if(i!=n-1)
			printf("\n");
	}
	return 0;
}