행렬의 곱셈

• 그냥 기초적인 행렬 곱셈 문제.
• 이론적으로는 슈트라센 알고리즘등을 사용하면 좀더 시간복잡도를 줄일수 있지만 (O(n^3)을 O(n^2.8)정도로..), n이 100정도의 범위에서는 별 의미가 없을 듯 해서 그냥 단순한 방법으로 구현.

"""Solution code for "Programmers 12949. 행렬의 곱셈".

- Problem link: https://programmers.co.kr/learn/courses/30/lessons/12949
- Solution link: http://www.teferi.net/ps/problems/programmers/12949
"""


def solution(arr1, arr2):
    answer = []
    for arr1_row in arr1:
        answer.append([(sum(x * y for x, y in zip(arr1_row, arr2_col))) 
                       for arr2_col in zip(*arr2)])
    return answer