蓝桥试题 算法提高 矩阵相乘 JAVA

问题描述
  小明最近在为线性代数而头疼,线性代数确实很抽象(也很无聊),可惜他的老师正在讲这矩阵乘法这一段内容。
  当然,小明上课打瞌睡也没问题,但线性代数的习题可是很可怕的。
  小明希望你来帮他完成这个任务。

现在给你一个ai行aj列的矩阵和一个bi行bj列的矩阵,
  要你求出他们相乘的积(当然也是矩阵)。
  (输入数据保证aj=bi,不需要判断)
输入格式
  输入文件共有ai+bi+2行,并且输入的所有数为整数(long long范围内)。
  第1行:ai 和 aj
  第2~ai+2行:矩阵a的所有元素
  第ai+3行:bi 和 bj
  第ai+3~ai+bi+3行:矩阵b的所有元素
输出格式
  输出矩阵a和矩阵b的积(矩阵c)
  (ai行bj列)
样例输入
2 2
12 23
45 56
2 2
78 89
45 56
样例输出
1971 2356
6030 7141

思路:输入数据的写法一定要写对,要不然就错掉了,俩数组交换相乘+=出来即可。

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int m = scanner.nextInt();
		int s = scanner.nextInt();
		int[][] a = new int[200][200];
		int[][] b = new int[200][200];
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < s; j++) {
				a[i][j] = scanner.nextInt();
			}
		}
		
		s = scanner.nextInt();               // 输入部分
		int n = scanner.nextInt();
		for (int i = 0; i < s; i++) {
			for (int j = 0; j < n; j++) {
				b[i][j] = scanner.nextInt();
			}
		}

		int[][] c = new int[200][200];
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < n; j++) {
				for (int k = 0; k < s; k++) {
					c[i][j] += a[i][k] * b[k][j];   // 相乘加等于给c数组
				}
			}
		}
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < n; j++) {
				System.out.print(c[i][j] + " ");       // 输出
			}
			System.out.println();
		}
	}
}

小剧场:你觉的我怎么样?

发布了201 篇原创文章 · 获赞 149 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/weixin_43771695/article/details/105275033