programmers - 행렬의 덧셈

이소라·2021년 8월 27일
0

문제 설명

  • 행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

제한 조건

  • 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

입출력 예

arr1arr2return
[[1,2],[2,3]][[3,4]],[5,6]][[4,6],[7,9]]
[[1],[2]][[3],[4]][[4],[6]]

문제 풀이

function solution(arr1, arr2) {
    var answer = [];
    for (let i = 0; i < arr1.length; i++) {
        let arr = [];
        for (let j = 0; j < arr1[i].length; j++) {
            arr.push(arr1[i][j] + arr2[i][j]);
        }
        answer.push(arr);
    }
    return answer;
}
  • 행렬 안에 행렬이 있으므로 for 문 2번 돌려줌
  • 빈 행렬 arr에 push하여 같은 행끼리 행렬로 묶어줌

다른 문제 풀이

function solution(arr1, arr2) {
  return arr1.map((row, rowIndex) => row.map((element, columnIndex) => element + arr2[rowIndex][columnIndex]));
}
  • map을 중복해서 사용할 수 있음

0개의 댓글