level 1. 행렬의 덧셈

출처

문제

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

예를 들어 2x2 행렬인 A = ((1, 2), (2, 3)), B = ((3, 4), (5, 6)) 가 주어지면, 같은 2x2 행렬인 ((4, 6), (7, 9))를 반환하면 됩니다. (어떠한 행렬에도 대응하는 함수를 완성해주세요.)

느낀 점

자바스크립트

• 2차원 배열을 만드는 것이 어려웠다. (answer[i] = [] 으로 빈 배열 속 배열을 만든다)
• .map() 메소드를 잘 활용하자. 참고
• 화살표 함수에 익숙해지자.참고

파이썬

• list comprehension을 활용하면 원하는 리스트 만들기가 정말 편하다.
• zip을 활용해서 2개 이상의 리스트를 활용해서 원하는 리스트를 만들 수 있다.
• 2차원 리스트를 만드는 것이 어려웠다.
  (answer = [[] for j in range(len(A))] 으로 리스트 A 길이만큼의 2차원 리스트를 만든다.)
• 리스트의의 길이만큼 반복하려면 for j in range(len(A)) 를 사용한다.
• 빈 2차원 리스트에 값을 넣으려면 a[0].append(1)를 사용한다. (a[0][0] = 1는 IndexError 발생)

풀이 (python, Javascript)

python - 풀이 1

def sumMatrix(A,B):
    answer = [[c + d for c, d in zip(a, b)] for a, b in zip(A,B)]
    return answer


# 아래는 테스트로 출력해 보기 위한 코드입니다.
print(sumMatrix([[1,2], [2,3]], [[3,4],[5,6]]))

python - 풀이 2

def sumMatrix(A,B):
    answer = [[] for j in range(len(A))]  
    for i in range(len(A)):
        for x in range(len(A[i])):
            answer[i].append(A[i][x] + B[i][x])
    return answer

# 아래는 테스트로 출력해 보기 위한 코드입니다.
print(sumMatrix([[1,2], [2,3]], [[3,4],[5,6]]))

Javascript

function sumMatrix(A,B){
	var answer = Array();
	for(var i = 0; i < A.length; i++){
		answer[i] = [];
			for(var j = 0; j < A[i].length; j++){
				answer[i][j] = A[i][j] + B[i][j];				
			}
		}
	return answer;
}

// 아래는 테스트로 출력해 보기 위한 코드입니다.
console.log(sumMatrix([[1,2], [2,3]], [[3,4],[5,6]]))

다른사람 풀이

python

def sumMatrix(A,B):
    answer = [[A[i][j] + B[i][j] for j in range(len(A[0]))] for i in range(len(A))]

    return answer

# 아래는 테스트로 출력해 보기 위한 코드입니다.
print(sumMatrix([[1,2], [3,4]], [[3,4],[5,6]]))

Javascript

function sumMatrix(A,B){
  return A.map((a,i) => a.map((b, j) => b + B[i][j]));
}

// 아래는 테스트로 출력해 보기 위한 코드입니다.
console.log(sumMatrix([[1,2], [2,3]], [[3,4],[5,6]]))

// 화살표 함수를 풀어 쓴 코드
function sumMatrix(A,B){
    return A.map(function(a, i){
             return a.map(function(b, j){
                     return b + B[i][j]
                   })
         })
}