2개의 행렬 arr1과 arr2를 입력 받아, 행렬 덧셈의 결과를 반환하는 함수 구현하기
행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1
과 arr2
를 입력 받아, 행렬 덧셈의 결과를 반환하는 함수, solution
을 완성해주세요.
arr1 | arr2 | return |
[[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 sum=[];
for(let j=0; j< arr1[0].length; j++){
sum.push(arr1[i][j]+arr2[i][j]);
}
answer.push(sum);
}
return answer;
}
def solution(arr1, arr2):
answer = []
for i in range(len(arr1)):
sum = []
for j in range(len(arr1[i])):
sum.append(arr1[i][j]+arr2[i][j])
answer.append(sum)
return answer
반복문 안에서 sum 이라는 배열을 추가로 생성했었는데, 다른 사람들의 풀이를 보고, answer
배열 하나만 사용해서 구현할 수 있다는 것을 배웠다.
function solution(arr1, arr2) {
var answer = [[]];
for(let i=0; i < arr1.length; i++){
answer[i]=[];
for(let j=0; j< arr1[0].length; j++){
answer[i].push(arr1[i][j]+arr2[i][j]);
}
}
return answer;
}