首页 > 试题广场 >

矩阵乘法

[编程题]矩阵乘法
  • 热度指数：86778 时间限制：C/C++ 1秒，其他语言2秒 空间限制：C/C++ 32M，其他语言64M
  • 算法知识视频讲解
\hspace{15pt}对于给定的 x 行 y 列的矩阵：
A = \begin{bmatrix} A_{1,1} & A_{1,2} & \cdots & A_{1,y} \\ A_{2,1} & A_{2,2} & \cdots & A_{2,y} \\ \vdots & \vdots & \ddots & \vdots \\ A_{x,1} & A_{x,2} & \cdots & A_{x,y} \end{bmatrix}
\hspace{15pt}和 y 行 z 列的矩阵：
B = \begin{bmatrix} B_{1,1} & B_{1,2} & \cdots & B_{1,z} \\ B_{2,1} & B_{2,2} & \cdots & B_{2,z} \\ \vdots & \vdots & \ddots & \vdots \\ B_{y,1} & B_{y,2} & \cdots & B_{y,z} \end{bmatrix}
\hspace{15pt}计算矩阵 A 和矩阵 B 的乘积 C。

\hspace{15pt}让我们回忆矩阵乘法的计算，对于第 i 行第 j 列的元素 C_{i,j} \left(1 \leqq i \leqq x;\ 1 \leqq j \leqq z\right)，有：
C_{i,j} = \sum\limits_{k=1}^{y} \left(A_{i,k} \times B_{k,j}\right)

输入描述:
\hspace{15pt}第一行输入一个整数 x \left(1 \leqq x \leqq 100\right) 代表第一个矩阵的行数。
\hspace{15pt}第二行输入一个整数 y \left(1 \leqq y \leqq 100\right) 代表第一个矩阵的列数和第二个矩阵的行数。
\hspace{15pt}第三行输入一个整数 z \left(1 \leqq z \leqq 100\right) 代表第二个矩阵的列数。
\hspace{15pt}此后 x 行，第 i 行输入 y 个整数 A_{i,1}, A_{i,2}, \dots, A_{i,y} \left(0 \leqq A_{i,j} \leqq 10\right) 代表矩阵 A 的第 i 行元素。
\hspace{15pt}此后 y 行，第 i 行输入 z 个整数 B_{i,1}, B_{i,2}, \dots, B_{i,z} \left(0 \leqq B_{i,j} \leqq 10\right) 代表矩阵 B 的第 i 行元素。


输出描述:
\hspace{15pt}输出 x 行，第 i 行输出 z 个整数，代表矩阵 C 的第 i 行元素。
示例1

输入

2
3
2
1 2 3
3 2 1
1 2
2 1
3 3

输出

14 13
10 11

说明

\hspace{15pt}在这个样例中，\begin{array}{ll}<br />C & = \begin{bmatrix} A_{1,1}B_{1,1} + A_{1,2}B_{2,1} + A_{1,3}B_{3,1} & A_{1,1}B_{1,2} + A_{1,2}B_{2,2} + A_{1,3}B_{3,2} \\ A_{2,1}B_{1,1} + A_{2,2}B_{2,1} + A_{2,3}B_{3,1} & A_{2,1}B_{1,2} + A_{2,2}B_{2,2} + A_{2,3}B_{3,2}\end{bmatrix} \\ <br />& = \begin{bmatrix} 1 \times 1 + 2 \times 2 + 3 \times 3 & 1 \times 2 + 2 \times 1 + 3 \times 3 \\ 3 \times 1 + 2 \times 2 + 1 \times 3 & 3 \times 2 + 2 \times 1 + 1 \times 3\end{bmatrix} \\<br />& = \begin{bmatrix} 1 + 4 + 9 & 2 + 2 + 9 \\ 3 + 4 + 1 & 6 + 2 + 3\end{bmatrix} \\<br />& = \begin{bmatrix} 14 & 13 \\ 10 & 11\end{bmatrix} \end{array}。
示例2

输入

16
8
7
17 19 16 19 14 1 14 9 
7 2 7 9 16 14 16 12 
13 3 3 17 5 9 8 16 
1 14 16 10 13 13 14 1 
13 13 15 4 7 2 6 16 
16 15 5 5 15 13 1 11 
11 5 0 16 14 7 7 15 
0 16 4 7 16 6 0 15 
2 14 11 2 17 17 5 12 
8 13 11 10 1 17 10 8 
15 16 17 15 7 8 13 14 
5 19 11 3 11 14 5 4 
9 16 13 11 15 18 0 3 
15 3 19 9 5 14 12 3 
9 8 7 11 18 19 14 18 
12 19 9 1 0 18 17 10 
5 18 16 19 6 12 5 
1 17 1 5 9 16 3 
14 16 4 0 19 3 6 
11 9 15 18 11 17 13 
5 5 19 3 16 1 12 
12 13 19 1 10 5 18 
19 18 6 18 19 12 3 
15 11 6 5 10 17 19

输出

1020 1490 1063 1100 1376 1219 884
966 1035 1015 715 1112 772 920
822 948 888 816 831 920 863
855 1099 828 578 1160 717 724
745 1076 644 595 930 838 688
635 1051 970 600 880 811 846
748 879 952 772 864 872 878
526 722 645 335 763 688 748
764 996 868 362 1026 681 897
836 1125 785 637 940 849 775
1082 1476 996 968 1301 1183 953
609 987 717 401 894 657 662
700 1083 1022 527 1016 746 875
909 1162 905 722 1055 708 720
1126 1296 1240 824 1304 1031 1196
905 1342 766 715 1028 956 749
算法知识视频讲解
添加回答
JavaScript V8
sirokuma2024头像 sirokuma2024 
while (line = readline()) {
    
    // 读取数据
    const x = parseInt(line, 10);
    const y = parseInt(readline());
    const z = parseInt(readline());
    const xyMatrix = [];
    const yzMatrix = [];
    const zyMatrix = [];
    for (let i = 0; i < x; i++) {
        xyMatrix[i] = readline().split(" ").map(Number);
    }
    for (let i = 0; i < y; i++) {
        yzMatrix[i] = readline().split(" ").map(Number);
    }
    for (let i = 0; i < z; i++) {
        const row = [];
        for (let j = 0; j < y; j++) {
            row.push(yzMatrix[j][i])
        }
        zyMatrix.push(row);
    }
    
    // 计算矩阵相乘的结果
    for (let i = 0; i < x; i++) {
        const row = [];
        for (let j = 0; j < z; j++) {
            const product = getProduct(xyMatrix[i], zyMatrix[j]);
            row.push(product);
        }
        console.log(...row);
    }
}
function getProduct(row, column) {
    let result = 0;
    for (let i = 0; i < row.length; i++) {
        result = result + (row[i] * column[i]);
    }
    return result;
}

发表于 2022-02-08 13:32:00 回复(0)
提交观点

问题信息

数组
难度：
1条回答 26935浏览

热门推荐

通过挑战的用户

查看代码

相关试题

矩阵乘法

扫一扫，把题目装进口袋

求职之前，先上牛客

扫描二维码，进入QQ群

扫描二维码，关注牛客公众号