蓝桥杯 算法训练 矩阵乘法 Java实现 避免运行超时。

算法训练 矩阵乘法

问题描述
　　输入两个矩阵，分别是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.*;

public class Main{
	public static void main(String[] args){
		Scanner scanner = new Scanner(System.in);
		int n = 0, m = 0, s = 0;
		m = scanner.nextInt();
		s = scanner.nextInt();
		n = scanner.nextInt();
		int[][] a = new int[m][s];
		int[][] b = new int[s][n];
		for(int i=0; i<m; i++){
			for(int j=0; j<s; j++){
				a[i][j] = scanner.nextInt();
			}
		}
		for(int i=0; i<s; i++){
			for(int j=0; j<n; j++){
				b[i][j] = scanner.nextInt();
			}
		}
		int[][] sum = new int[m][n];
		for(int i=0; i<m; i++){
			for(int j=0; j<n; j++){
				int temp = 0;
				for(int k=0; k<s; k++){
					temp += a[i][k]*b[k][j];
				}
				sum[i][j] = temp;
			}
		}
		
		for(int i=0; i<m; i++){
			for(int j=0; j<n; j++){
				System.out.print(sum[i][j]+" ");
			}
			System.out.println();
		}
	}
}

这个程序提交上去只能得85分，因为运行超时。主要问题应该是输入输出超时。用scanner输入太慢了。下面进行改进，使用BufferedReader来读取数据。具体代码如下：

解法二：

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

class Reader {
	static BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
	static StringTokenizer tokenizer = new StringTokenizer("");
	static String nextLine() throws IOException{// 读取下一行字符串
		return reader.readLine();
	}
	static String next() throws IOException {// 读取下一个字符串
		while (!tokenizer.hasMoreTokens()) {
			tokenizer = new StringTokenizer(reader.readLine());
		}
		return tokenizer.nextToken();
	}

	static int nextInt() throws IOException {// 读取下一个int型数值
		return Integer.parseInt(next());
	}

	static double nextDouble() throws IOException {// 读取下一个double型数值
		return Double.parseDouble(next());
	}
}


public class Main{
	public static void main(String[] args) throws IOException{
		Reader reader = new Reader();
		Scanner scanner = new Scanner(System.in);
		int n = 0, m = 0, s = 0;
		m = Reader.nextInt();
		s = Reader.nextInt();
		n = Reader.nextInt();
		int[][] a = new int[m][s];
		int[][] b = new int[s][n];
		for(int i=0; i<m; i++){
			for(int j=0; j<s; j++){
				a[i][j] = Reader.nextInt();
			}
		}
		for(int i=0; i<s; i++){
			for(int j=0; j<n; j++){
				b[i][j] = Reader.nextInt();
			}
		}
		int[][] sum = new int[m][n];
		for(int i=0; i<m; i++){
			for(int j=0; j<n; j++){
				int temp = 0;
				for(int k=0; k<s; k++){
					temp += a[i][k]*b[k][j];
				}
				sum[i][j] = temp;
			}
		}
		
		for(int i=0; i<m; i++){
			for(int j=0; j<n; j++){
				System.out.print(sum[i][j]+" ");
			}
			System.out.println();
		}
	}
}

这样提交上去就不会超时了。参加OJ竞赛是经常会遇到大量的数据进行输入输出。建议可以把程序中Reader类给记下来，比赛时用Reader代替Scanner。
其中nextLine对应Scanner.nextLine();

next() 对应Scanner.next();

nextInt() 对应Scanner.nextInt() ;

nextDouble() 对应Scanner.nextDouble() ;
这样可以避免出现超时的情况。