矩阵乘法

最新推荐文章于 2025-10-10 19:48:18 发布

原创最新推荐文章于 2025-10-10 19:48:18 发布 · 415 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

20 篇文章

订阅专栏

博客内容介绍了矩阵乘法的算法实现，并针对输入数据量较大导致的运行超时问题，提出了使用替代输入流方法来优化Java代码，解决了在蓝桥杯竞赛中遇到的性能问题。

算法地址：http://lx.lanqiao.cn/problem.page?gpid=T218
问题描述
　　输入两个矩阵，分别是ms，sn大小。输出两个矩阵相乘的结果。
输入格式
　　第一行，空格隔开的三个正整数m,s,n（均不超过200）。
　　接下来m行，每行s个空格隔开的整数，表示矩阵A（i，j）。
　　接下来s行，每行n个空格隔开的整数，表示矩阵B（i，j）。
输出格式
　　m行，每行n个空格隔开的整数，输出相乘後的矩阵C（i，j）的值。
样例输入
2 3 2
1 0 -1
1 1 -3
0 3
1 2
3 1
样例输出
-3 2
-8 2
提示
矩阵C应该是m行n列，其中C(i,j)等于矩阵A第i行行向量与矩阵B第j列列向量的内积。
例如样例中C(1,1)=(1,0,-1)(0,1,3) = 1 * 0 +01+(-1)*3=-3

矩阵的乘法相信大家都会吧，我就不仔细讲了，直接把我写的代码粘出来

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int m,s,n;
		m=sc.nextInt();
		s=sc.nextInt();
		n=sc.nextInt();
		int[][] jz1 = new int[m][s];
		int[][] jz2 = new int[s][n];
		int[][] res = new int[m][n];
		for(int i=0;i<jz1.length;i++) {
			for(int j=0;j<jz1[0].length;j++) {
				jz1[i][j] = sc.nextInt();
			}
		}
		for(int i=0;i<jz2.length;i++) {
			for(int j=0;j<jz2[0].length;j++) {
				jz2[i][j] = sc.nextInt();
			}
		}
		for(int i=0;i<res.length;i++) {
			for(int j=0;j<res[0].length;j++){
				for(int o=0;o<jz1[i].length;o++) {
					res[i][j]+=jz1[i][o]*jz2[o][j];
				}
				if(j!=res[0].length-1)
					System.out.print(res[i][j]+" ");
				else
					System.out.print(res[i][j]);
			}
			System.out.println();
		}
	}
}

写出来以后，就到蓝桥杯官网上跑去了，但是出来最后一个运行超时。
算法感觉是没什么优化的地方了，所以我就感觉输入有问题，因为矩阵输入的数据量比较大，但是 Java 自带的 Scanner 方法很耗费内存，所以就用了另一种输入流的方式。

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.io.StreamTokenizer;

public class Main {
	static Reader reader;
	public static void main(String[] args) {
		StreamTokenizer in = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in))); // 用于输入的对象
		int[][] jz1 = null;
		int[][] jz2 = null;
		int[][] res = null;
		try {
			in.nextToken();
			int m,s,n;
			m=(int)in.nval;
			in.nextToken();
			s=(int)in.nval;
			in.nextToken();
			n=(int)in.nval;
			jz1 = new int[m][s];
			jz2 = new int[s][n];
			res = new int[m][n];
			for(int i=0;i<jz1.length;i++) {
				for(int j=0;j<jz1[0].length;j++) {
					in.nextToken();
					jz1[i][j] = (int)in.nval;
				}
			}
			for(int i=0;i<jz2.length;i++) {
				for(int j=0;j<jz2[0].length;j++) {
					in.nextToken();
					jz2[i][j] = (int)in.nval;
				}
			}
		} catch (IOException e) {
			e.printStackTrace();
		} // 必须先调用这个把下一个数据拉进来
		for(int i=0;i<res.length;i++) {
			for(int j=0;j<res[0].length;j++){
				for(int o=0;o<jz1[i].length;o++) {
					res[i][j]+=jz1[i][o]*jz2[o][j];
				}
				if(j!=res[0].length-1)
					System.out.print(res[i][j]+" ");
				else
					System.out.print(res[i][j]);
			}
			System.out.println();
		}
	}
}

这次的代码在去跑就不会出现超时的问题了！