ABOUT ME

-

Today
-
Yesterday
-
Total
-
  • 29. 행렬의 곱셈
    코딩 테스트/Level 2 2020. 8. 13. 10:03
    반응형

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

    행렬의 곱셈
    연습문제
    3222명 완료

    종이 펴놓고 그려보면 
    어렵지 않게 풀 수 있는 문제...

    파이썬..

    def solution(arr1, arr2):
        answer = []
        for idx1 in range(len(arr1)):
            row = []
            for idx2 in range(len(arr2[0])):
                tmp = 0
                for idx3 in range(len(arr1[0])):
                    tmp += arr1[idx1][idx3] * arr2[idx3][idx2]
                row.append(tmp)
            answer.append(row)
        return answer

    Java

    import java.util.ArrayList;
    
    class Solution {
        public ArrayList<ArrayList<Integer>> solution(int[][] arr1, int[][] arr2) {
            var answer = new ArrayList<ArrayList<Integer>>();
            for (var i = 0; i < arr1.length; i++) {
                var row = new ArrayList<Integer>();
                for (var j = 0; j < arr2[0].length; j++) {
                    var tmp = 0;
                    for (var k = 0; k < arr1[0].length; k++) {
                        tmp += arr1[i][k] * arr2[k][j];
                    }
                    row.add(tmp);
                }
                answer.add(row);
            }
            return answer;
        }
    }
    반응형
Designed by Tistory.