快蓝桥杯了,好久没刷题,连这个都不会了,还得看网上的资料,这几天写博客记录每天刷的算法题,奥利给!
标题
题目
资源限制
时间限制: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;
}