描述
给定两个N × N的稀疏矩阵A和B,其中矩阵A有P个元素非0,矩阵B有Q个元素非0。请计算两个矩阵的乘积C = A × B并且输出C中所有非0的元素。
输入
第一行包含三个整数N, P, Q
以下P行每行三个整数i, j, k表示A矩阵的一个非0元素:Aij = k
以下Q行每行三个整数i, j, k表示B矩阵的一个非0元素:Bij = k
对于80%的数据,1 ≤ N, P, Q ≤ 200
对于100%的数据, 1 ≤ N, P, Q ≤ 2000, 1 ≤ i, j ≤ N, 0 ≤ k ≤ 100
输出
输出若干行,按先行后列的顺序输出矩阵C的每一个非0元素
每行三个整数i, j, k表示C矩阵的一个非0元素:Cij = k
样例输入
2 2 4
1 1 1
2 2 1
1 1 1
1 2 2
2 1 3
2 2 4
样例输出
1 1 1
1 2 2
2 1 3
2 2 4
解析
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
int N = sc.nextInt();
int P = sc.nextInt();
int Q = sc.nextInt();
int[][] A = new int[P][3];
int[][] B = new int[Q][3];
for (int i = 0; i < P; i++) {
for (int j = 0; j < 3; j++) A[i][j] = sc.nextInt();
}
for (int i = 0; i < Q; i++) {
for (int j = 0; j < 3; j++) B[i][j] = sc.nextInt();
}
int[][] C = new int[N ][N];
for (int i = 0; i < P; i++) {
for (int j = 0; j < Q; j++) {
if (A[i][1] == B[j][0]) {//由图可以看出当A矩阵的列号等于B矩阵的行号时求和
C[A[i][0]-1][B[j][1]-1] += A[i][2] * B[j][2];
}
}
}
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
if (C[i][j] == 0) continue;
System.out.println((i+1)+ " " + (j+1) + " " + C[i][j]);
}
}
}
}