1230: 矩阵乘法

题目

Description

矩阵乘法是线性代数中最基本的运算之一。
给定三个矩阵 A\B\C 请编写程序判断A*B = C 是否成立。
Input

输入包含多组数据。每组数据的格式如下：
第一行包括两个整数p 和q，表示矩阵A 的大小。后继p 行，每行有q 个整数，表示矩阵A
的元素内容。
紧接着用相同的格式给出矩阵B 和矩阵C。
输入数据的最后一行是两个0，你的程序处理到这里时就应该退出了。
输入数据中所有的整数绝对值都不超过100。
Output

对每一组输入数据，你的程序都要输出单独一行字符。
如果 A*B=C成立则输出”Yes”
如果 A 和B 根本不能相乘，或A*B=C不成立，则输出”No”。注意大小写。
Sample Input

2 3
1 2 3
4 5 6
3 2
1 2
3 4
5 6
2 2
22 28
49 64
1 2
1 2
2 1
2
1
2 2
2 4
1 2
1 2
1 2
1 1
1
1 1
1

0 0
Sample Output

Yes
No
No

---

代码块

import java.util.Arrays;import java.util.Scanner;public class Main {    public static void main(String[] args) {        Scanner cn = new Scanner(System.in);        while (cn.hasNext()) {            int pA = cn.nextInt();            int qA = cn.nextInt();            if(pA==0&&qA==0) return;            int[][] a = new int[pA][qA];            for (int i = 0; i < pA; i++)                for (int j = 0; j < qA; j++)                    a[i][j] = cn.nextInt();            int pB = cn.nextInt();            int qB = cn.nextInt();            int[][] b = new int[pB][qB];            for (int i = 0; i < pB; i++)                for (int j = 0; j < qB; j++)                    b[i][j] = cn.nextInt();            int pC = cn.nextInt();            int qC = cn.nextInt();            int[][] c = new int[pC][qC];            for (int i = 0; i < pC; i++)                for (int j = 0; j < qC; j++)                    c[i][j] = cn.nextInt();            if (qA == pB) {                if (qC == qB && pC == pA) {                    int[][] as = new int[pC][qC];                    for (int i = 0; i < pC; i++) {                        for (int j = 0; j < qC; j++) {                            as[i][j] = 0;                        }                    }                    int z[][] = new int[a.length][b[0].length];                    for (int i = 0; i < a.length; i++) {                        for (int j = 0; j < b[0].length; j++) {                            int temp = 0;                            for (int x = 0; x < b.length; x++) {                                temp += a[i][x] * b[x][j];                            }                            z[i][j] = temp;                        }                    }                    boolean flag = true;                    for (int i = 0; i < pC; i++) {                        for (int j = 0; j < qC; j++) {                            if(z[i][j]!=c[i][j]) {                                System.out.println("No");                                flag = false;                                break;                            }                        }                        if(!flag) break;                    }                    if(flag) System.out.println("Yes");                } else                    System.out.println("No");            } else                System.out.println("No");        }        cn.close();    }}