hihoCoder1675蓝桥算法基础第四章测验题目4稀疏矩阵乘积Java代码

描述
给定两个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]);
            }
        }
	}
}