-
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; } }
반응형