【蓝桥】基础练习 矩阵乘法

核能气质少年 提交于 2020-02-02 20:36:37

原文地址
问题描述
  给定一个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<iostream>
using namespace std;

int main()
{
	int N,M,a[30][30],b[30][30],c[30][30];  // 输入a:  c=0  c=b*a  b=c  c=0  b为最终结果 
 	cin>>N>>M;
	
	for(int i=0;i<N;i++){      // 输入a矩阵元素 
		for(int j=0;j<N;j++){
			cin>>a[i][j];
			b[i][j] = a[i][j];
			c[i][j] = 0;
		}
	}
	
	if(M==0){     // 0次方输出单位矩阵 ,用 c矩阵另外输出 
	    for(int i=0;i<N;i++){
	    	for(int j=0;j<N;j++){
	    		if(i==j)
	    		    c[i][j]=1;
	    		cout<<c[i][j]<<" ";
	    	}
	    	cout<<endl;
	    } 
	    return 0;   // 结束程序 
    }
	    
    while(--M){                     // 乘的次数为 M-1 
		for(int i=0;i<N;i++){       // 一次c矩阵结果 
			for(int j=0;j<N;j++){
			   for(int k=0;k<N;k++){
			       c[i][j] += a[i][k]*b[k][j];
		       }    	
			}
		} 
		for(int i=0;i<N;i++){        
			for(int j=0;j<N;j++){
			    b[i][j] = c[i][j]; //  b保存结果 
				c[i][j] = 0;   	
			}
		} 
    }
    
    for(int i=0;i<N;i++){         // 输出结果 
    	for(int j=0;j<N;j++){
    		cout<<b[i][j]<<" "; 
    	}
    	cout<<endl;
    } 
    
	return 0;
}

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!