[백준/python] 2740 행렬곱셈

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

 

단순 행렬곱셈 문제이다. 우리에겐 넘파이가 있지만... 백준에선 직접 곱해주어야 한다.

def main(): 
    n, _ = map(int,input().split())
    arr = []
    for _ in range(n):
        arr.append(list(map(int,input().split())))
    m, k = map(int,input().split())
    brr = []
    for _ in range(m):
        brr.append(list(map(int,input().split())))
    
    ans = [[[0] for _ in range(k)] for i in range(n)]
    
    for i in range(n):
        for j in range(k):
            sum = 0
            for x in range(m):
                sum += arr[i][x]*brr[x][j]
            ans[i][j] = sum

    for i in range(n):
        print(' '.join(map(str,ans[i])))

if __name__ == "__main__":
    main()

너무너무 편안한 브루트포스문제