알고리즘 공부[Javascript]/프로그래머스 2단계

행렬의 곱셈

Kevinkb 2021. 7. 5. 22:19

문제 설명

2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

제한 조건

  • 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
  • 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
  • 곱할 수 있는 배열만 주어집니다.

입출력 예

arr1 arr2 return
[[1, 4], [3, 2], [4, 1]] [[3, 3], [3, 3]] [[15, 15], [15, 15], [15, 15]]
[[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]]
function solution(arr1, arr2) {
    var answer = [];
    let arr = [];
    let sum = 0;
    
    for (let k = 0; k < arr1.length; k++) {
        for (let j = 0; j < arr2[0].length; j++) {
            for (let i = 0; i < arr1[k].length; i++) {
                sum += (arr1[k][i] * arr2[i][j]);
            }
            arr.push(sum);
            sum = 0;
        }
        answer.push(arr);
        arr = [];
    }
    return answer;
}

// map(), reduce()
function solution(arr1, arr2) {
    return arr1.map((v) => arr2[0].map((x,y) => v.reduce((a,b,c) => a + b * arr2[c][y], 0)))
}

 

'알고리즘 공부[Javascript] > 프로그래머스 2단계' 카테고리의 다른 글

최솟값 만들기  (0) 2021.07.06
피보나치 수  (0) 2021.07.06
JadenCase 문자열 만들기  (0) 2021.07.05
N개의 최소공배수  (0) 2021.07.04
H-index  (0) 2021.07.02