基础训练6⭐-矩阵乘法

矩阵乘法

问题描述

　　给定一个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

 

一开始看这道题，有点伤脑筋，假设用太多的循环，则会造成时间超时，特别是用java的情况下。

两个矩阵相乘，怎么相乘呢？

 

计算的时候要特别注意n阶矩阵的0次幂等于n阶单位矩阵。

 

Java实现：

import java.util.Scanner;

public class Main{
	
	public static void printMatrix(int[][] intArr){
		int length = intArr.length;
		for (int i = 0; i < length; i++){
			for (int j = 0; j < length; j++){
				System.out.print(intArr[i][j] + " ");
			}
			System.out.println();
		}
	}
	
	public static int[][] powerMatrix(int[][] intResult, int[][] intArr, int n){
		int num;
		int[][] res = new int[n][n];
		for (int i = 0; i < n; i++){
			for (int j = 0; j < n; j++){
				num = 0;
				for (int k = 0; k < n; k++)
					num += intResult[k][j] * intArr[i][k];
				res[i][j] = num;
			}
		}
		return res;
	}
	
	public static void main(String[] args){
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int M = sc.nextInt();
		int[][] intArr = new int[N][N];
		int[][] intResult = new int[N][N];
		for (int i = 0; i < N; i++)
			for (int j = 0; j < N; j++)
				intResult[i][j] = intArr[i][j] = sc.nextInt();
			for (int i = 1; i < M; i++)	
			intResult = powerMatrix(intResult,  intArr, N);
			if (M == 0)
				for (int i = 0; i < N; i++)
					for (int j = 0; j < N; j++){
						if (i != j)
							intResult[i][j] = 0;
						else
							intResult[i][j] = 1;
					}
		printMatrix(intResult);
	}
}

总结：利用递归返回一个二维数组容易实现幂计算。

C++实现：

#include <stdio.h>
#include <string.h>
#define N 100
int A[N][N],t[N][N],r[N][N];
int main()
{
	int n,m,i,j,k;
	scanf("%d%d",&n,&m);
	for(i=0;i<n;i++)
		for(j=0;j<n;j++)
			scanf("%d",&A[i][j]);
	for(i=0;i<n;i++) 
 		r[i][i]=1;//单位矩阵，如同数的乘法中的1
	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]+=r[i][k]*A[k][j];
		for(i=0;i<n;i++)
			for(j=0;j<n;j++)
		 		r[i][j]=t[i][j];
	}
	for(i=0;i<n;i++)
	{
		for(j=0;j<n-1;j++)
			printf("%d ",r[i][j]);	
		printf("%d\n",r[i][j]);	
	}
	return 0;
}