파이썬) 1일 1문제 또는 2문제 알고리즘 풀기(프로그래머스)

5월 21일 월요일

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

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

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

view raw matrix hosted with ❤ by GitHub

행렬문제는 행렬 이라는 압박감을 벗어나느게 우선인듯. 처음에 예시 문제가 2*2 여서 여기서 함정에 빠졌다. 단순한 로직이지만 중요한 점은 행, 열을 구분해야한다는 것. 다른 분들의 풀이도 비슷하지만 zip(A,B) 함수를 이용한게 신기하였음.