Java实现 稀疏矩阵乘积

爷,独闯天下 提交于 2020-02-25 20:06:56

稀疏矩阵乘积

描述
给定两个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

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]);
			}
			 
		}
	}

}

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!