문제: 행렬의 덧셈
문제 요약
주어진 두 개의 행렬의 같은 행, 같은 열의 값을 더하기
입출력 예시
arr1 | arr2 | return |
---|---|---|
[[1,2],[2,3]] | [[3,4],[5,6]] | [[4,6],[7,9]] |
[[1],[2]] | [[3],[4]] | [[4],[6]] |
풀이(Javascript)
두 개의 배열이 반복되어야 하므로, 이중 for
문을 사용해서 구현했다.
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;
}
반복문 안에서 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;
}
풀이(Python)
def solution(arr1, arr2):
answer = []
for i in range(len(arr1)):
sum = []
for j in range(len(arr1[0])):
sum.append(arr1[i][j]+arr2[i][j])
answer.append(sum)
return answer