BASIC-17 矩阵乘法

问题描述

基础练习 矩阵乘法

时间限制：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

解题思路

//首先要声明一个新矩阵，用来存储矩阵相乘的结果
//结果矩阵的第i,j位置上的结果由A矩阵的第i行与B矩阵的第j列各个元素的相乘最后相加获得。
//A矩阵的行(行固定，变动的是i行中的列) * B矩阵的列(列固定，变动的是j列中的行)
//建立一个for循环，来表示变动的行或列
//当次幂为0时，将从左上角到右下角对角线值变成1

核心代码

//复制一个相同的矩阵，当作被乘矩阵
long[][] temp = arr;
//进行幂运算
for (int i = 1; i < pow; i++) {
	//矩阵相乘后结果返回给原矩阵，原矩阵继续乘被乘矩阵。（完成幂运算）
	arr = arr_pow(arr , temp);
}

//建立一个for循环，来表示变动的行或列
for (int d = 0; d < result.length; d++) {
	result[i][j] += arr[i][d] * temp[d][j];
}

	//0次幂运算
	if (pow == 0) {
		//清空原矩阵中所有值，将从左上角到右下角对角线值变成1
		long[][] result = new long[n][n];
		for (int i = 0; i < result.length; i++)
			result[i][i] = 1;
		arr = result;
	}

参考代码

package BASIC_17;

import java.util.Scanner;

public class Main {
public static void main(String[] args) {
	Scanner sr = new Scanner(System.in);
	int n = sr.nextInt();
	int pow = sr.nextInt();
	long[][] arr = new long[n][n];
	//创建矩阵并赋值
	for (int i = 0; i < arr.length; i++)
		for (int j = 0; j < arr[0].length; j++)
			arr[i][j] = sr.nextLong();
	//当次幂为0时，将从左上角到右下角对角线值变成1
	//0次幂运算
	if (pow == 0) {
		//清空原矩阵中所有值，将从左上角到右下角对角线值变成1
		long[][] result = new long[n][n];
		for (int i = 0; i < result.length; i++)
			result[i][i] = 1;
		arr = result;
	}
	//复制一个相同的矩阵，当作被乘矩阵
	long[][] temp = arr;
	//进行幂运算
	for (int i = 1; i < pow; i++) {
		//矩阵相乘后结果返回给原矩阵，原矩阵继续乘被乘矩阵。（完成幂运算）
		arr = arr_pow(arr , temp);
	}
	//输出结果
	for (int i = 0; i < arr.length; i++) {
		for (int j = 0; j < arr.length; j++) {
			System.out.print(arr[i][j]+" ");
		}System.out.println();
	}
}
//矩阵乘法
private static long[][] arr_pow(long[][] arr, long[][] temp) {
	long[][] result = new long[arr.length][arr[0].length];
	for (int i = 0; i < result.length; i++) {
		for (int j = 0; j < result[0].length; j++) {

			//建立一个for循环，来表示变动的行或列
			for (int d = 0; d < result.length; d++) {
				result[i][j] += arr[i][d] * temp[d][j];
			}
		}
	}
	return result;
}
}