본문 바로가기

PS/프로그래머스

[프로그래머스 LV 2 ] 행렬의 곱셈 파이썬/python

728x90

https://programmers.co.kr/learn/courses/30/lessons/12949#qna

 

코딩테스트 연습 - 행렬의 곱셈

[[2, 3, 2], [4, 2, 4], [3, 1, 4]] [[5, 4, 3], [2, 4, 1], [3, 1, 1]] [[22, 22, 11], [36, 28, 18], [29, 20, 14]]

programmers.co.kr

 

말 그대로 행렬의 곱셈을 구현했다.

프로그래머스 다른 사람들 풀이 보니 넘파이로 간단하게 풀기도하고

알아보지도 못 하겠는 엄청 짧은 코드로 풀고 그랬던데..

 

사실 그렇게 풀 실력도안되고 코테볼때 생각도 안 날거같음

주먹구구식풀이법임

 

행렬의 곱셈 법은 위 사진을 보면 고등학교때 배웠던거 생각나실듯

 

 

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
def solution(arr1, arr2):
    answer = []
 
    for row in range(len(arr1)):
        li = []
        for ar2 in range(len(arr2[0])):
            sum = 0
            for col in range(len(arr1[0])):
                mul = arr1[row][col] * arr2[col][ar2]
                sum += mul
            li.append(sum)
        answer.append(li)
 
    return answer
 
 
cs
728x90