【华为机试069】矩阵乘法

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/HEYIAMCOMING/article/details/81117017

题目描述:

如果A是个x行y列的矩阵,B是个y行z列的矩阵,把A和B相乘,其结果将是另一个x行z列的矩阵C。

Java实现:

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        while (sc.hasNext()) {
            int x = sc.nextInt();
            int y = sc.nextInt();
            int z = sc.nextInt();
            int[][] A = new int[x][y];
            int[][] B = new int[y][z];
            int[][] C = new int[x][z];
            for (int i = 0; i < x; i++) {
                for (int j = 0; j < y; j++) {
                    A[i][j] = sc.nextInt();
                }
            }
            for (int i = 0; i < y; i++) {
                for (int j = 0; j < z; j++) {
                    B[i][j] = sc.nextInt();
                }
            }

            for (int i = 0; i < x; i++) {
                for (int j = 0; j < z; j++) {
                    C[i][j] = helper(A, B , i, j, y);
                    if (j != z-1)
                        System.out.print(C[i][j] + " ");
                    else 
                        System.out.print(C[i][j]);
                }
                System.out.println();
            }


        }
    }

    private static int helper(int[][] A, int[][] B, int i, int j, int y) {
        int sum = 0;
        for (int k = 0; k < y ; k++) {
            sum += A[i][k] * B[k][j];
        }
        return sum;
    }


}

知识点:

  • 需要注意x,y,z三者之间的关系

猜你喜欢

转载自blog.csdn.net/HEYIAMCOMING/article/details/81117017