boj2740 - 행렬곱셈

문제 : https://www.acmicpc.net/problem/2740

코드

import sys

if __name__ == '__main__':
    input = sys.stdin.readline

    row1, col1 = map(int, input().split())
    mat1 = [list(map(int, input().split())) for i in range(row1)]
    row2, col2 = map(int, input().split())
    mat2 = [list(map(int, input().split())) for i in range(row2)]
    result = [[0 for _ in range(col2)] for _ in range(row1)]
    # print(result)
    for i in range(row1):
        for j in range(col2):
            for k in range(col1):
                result[i][j] += mat1[i][k]*mat2[k][j]

    for i in range(row1):
        for j in range(col2):
            print(result[i][j], end = ' ')
        print()

그냥 정석적인 행렬 곱으로 품