基础训练-矩阵乘法

资源限制
时间限制: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部分时数据不能ac;非得n加一个数在这里插入图片描述

在这里插入图片描述

#include <bits/stdc++.h>
int main(){
    
    
	int n,m,i,j,k;
	scanf("%d%d",&n,&m);
	int a[n+1][n+1],t[n+1][n+1],d[n+1][n+1];
	for(i=0;i<n;i++){
    
    
		d[i][i]=1;//单位矩阵
	}
	for(i=0;i<n;i++){
    
    
		for(j=0;j<n;j++){
    
    
			scanf("%d",&a[i][j]);
		}
	}
	while(m--){
    
    
		memset(t,0,sizeof(t));
		for(i=0;i<n;i++){
    
    
			for(j=0;j<n;j++){
    
    
				for(k=0;k<n;k++){
    
    
					t[i][j]+=d[i][k]*a[k][j];
				}
			}
		}
	    for(i=0;i<n;i++){
    
    
	    	for(j=0;j<n;j++){
    
    
	    		d[i][j]=t[i][j];
			}
		}	
		
	}
	for(i=0;i<n;i++){
    
    
		for(j=0;j<n-1;j++){
    
    
	printf("%d ",d[i][j]);}
	printf("%d\n",d[i][j]);
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43959743/article/details/113006209