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

package 第二次模拟;

import java.util.Scanner;

public class Demo4矩阵相乘 {
	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();
			}
		}
		sc.close();
		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++) {
				System.out.println(i+1+" "+(j+1)+" "+c[i][j]);
			}

		}
	}

}