华为机试HJ69矩阵乘法

题目：

输入：
2
3
2
1 2 3
3 2 1
1 2
2 1
3 3
输出：
14 13
10 11
说明：
1 2 3
3 2 1
乘以
1 2
2 1
3 3
等于
14 13
10 11
输入的第一行为第一个矩阵的行数，第二行为第一个矩阵的列数或第二个矩阵的行数，第三行为第二个矩阵的列数，完成两个矩阵相乘。

想法：

按第一行显示的第一个矩阵的行数，将第一个矩阵与第二个矩阵取出，再将其中对应元素相乘。

import sys

M = []  # 第一个矩阵
N = []  # 第二个矩阵
m_r = 0  # 第一个矩阵的行数
m_c = 0  # 第一个矩阵的列数或第二个矩阵的行数
n_r = 0  # 第二个矩阵的列数
count = 1
for line in sys.stdin:
    a = line.split()  # 按行读取数据
    a = list(map(int, a))  # 将字符串转为数值列表
    # a = [int(x) for x in a]
    # 按行获取行列数和矩阵
    if count == 1:
        m_r = a[0]
    elif count == 2:
        m_c = a[0]
    elif count == 3:
        n_r = a[0]
    elif count >= 4 and count <= 3+m_r:
        M.append(a)
    elif count > 3+m_r:
        N.append(a)
    count += 1

# 实现矩阵相乘
MN = [[sum(M[i][k]*N[k][j] for k in range(m_c)) for j in range(n_r)] for i in range(m_r)]

# 按行输出结果
for l in MN:
    print(' '.join(map(str, l)))