2738: 행렬 덧셈 - Python

https://www.acmicpc.net/problem/2738

문제

N*M크기의 두 행렬 A와 B가 주어졌을 때, 두 행렬을 더하는 프로그램을 작성하시오.

입력

첫째 줄에 행렬의 크기 N 과 M이 주어진다. 둘째 줄부터 N개의 줄에 행렬 A의 원소 M개가 차례대로 주어진다. 이어서 N개의 줄에 행렬 B의 원소 M개가 차례대로 주어진다. N과 M은 100보다 작거나 같고, 행렬의 원소는 절댓값이 100보다 작거나 같은 정수이다.

출력

첫째 줄부터 N개의 줄에 행렬 A와 B를 더한 행렬을 출력한다. 행렬의 각 원소는 공백으로 구분한다.

예제 입출력

풀이 1 (정답)

N, M = map(int, input().split())
mat1, mat2 = [], []

for row in range(N):
	row = list(map(int, input().split()))
	mat1.append(row)

for row in range(N):
	row = list(map(int, input().split()))
	mat2.append(row)

for row in range(N):
	for col in range(M):
		print(mat1[row][col] + mat2[row][col], end=' ')
	print()

ImJay님의 코드를 참고했다.
2차원 배열을 사용해본 적이 없어 어려웠다.

풀이 2 (정답, 풀이 1 수정)

N, M = map(int, input().split())
mat1, mat2 = [], []

for _ in range(N):
	row = list(map(int, input().split()))
	mat1.append(row)

for _ in range(N):
	row = list(map(int, input().split()))
	mat2.append(row)

for row in range(N):
	for col in range(M):
		print(mat1[row][col] + mat2[row][col], end=' ')
	print()

풀이 1에서 mat1, mat2를 입력받는 for문에서 반복변수를 'row'로 썼는데,
다시 보니 어색해서 '_'로 고쳐보았더니 역시 잘 돌아갔다.

또한 C, Java 등의 언어로 문제를 풀 때와는 달리,
Python에서는 입력받은 열 개수(M)는 행렬을 입력받는 과정에서는 큰 의미가 없다.
이는 열 데이터를 개수 제한이 없는 input().split()으로 입력받기 때문이다.

풀이 3 (정답, unpacking)

N, M = map(int, input().split())
mat1, mat2 = [], []

for _ in range(N):
    mat1.append(list(map(int, input().split())))
for _ in range(N):
    mat2.append(list(map(int, input().split())))

for row in range(N):
    mat3 = []
    for col in range(M):
        mat3.append(mat1[row][col] + mat2[row][col])
    print(*mat3)

<백준 티어별 풀이노트>를 참고하였다.
언패킹에 익숙해지고 싶은데 문제 풀 때 잘 안떠오른다.
의식해서 연습해봐야겠다.

오늘의 교훈

• 행렬을 입력받는 문제에서는 아래의 순서로 접근해보자.

  1. 우선 행렬을 빈 리스트로 선언한다. e.g. mat = []
  2. 행에 대해 입력받는다. e.g. row = list(map(int, input().split()
  3. 입력받은 행을 추가하여 행렬을 완성한다. e.g. mat.append(row)

• 행렬의 값에 반복하여 접근할 때는 이중 for문을 쓰고, for row -> for col의 순서로 접근한다.