[백준] 2738번 파이썬 python(행렬 덧셈)

▷ 문제

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

 

▷ 전체 코드

N, M = map(int, input().split())
a = []
b = []

for i in range(N):
    a.append(list(map(int, input().split())))

for i in range(N):
    b.append(list(map(int, input().split())))

for i in range(N):
    for j in range(M):
        print(a[i][j] + b[i][j], end=' ')
    print()

 

▷ 코드 분석

N, M = map(int, input().split())
a = []
b = []

변수 N과 M에 각각 행렬의 행과 열의 수를 저장한다.

map() 함수를 사용해서 입력된 요소에 지정한 함수를 적용한다. int() 함수를 사용해 저장된 문자열을 정수형으로 변환한다.

input() 함수를 사용해 입력 받은 요소를 변수에 문자열로 저장한다. split() 함수를 사용해 입력 받은 요소를 공백을 기준으로 나눈다.

 

빈 리스트 a와 b를 만든다.

 

for i in range(N):
    a.append(list(map(int, input().split())))

for i in range(N):
    b.append(list(map(int, input().split())))

for 문을 사용해 행의 수만큼 반복하는 반복문을 만든다.

 

append() 함수를 사용해 빈 리스트에 입력 받은 리스트를 저장한다.

list() 함수를 사용해 괄호 안에 요소를 리스트에 저장한다. map() 함수를 사용해 입력된 요소에 지정한 함수를 적용한다.

input() 함수를 사용해 입력 받은 요소를 문자열로 변수에 저장한다. split() 함수를 사용해 입력 받은 요소를 공백을 기준으로 나눈다.

int() 함수를 사용해 저장된 문자열을 정수형으로 변환한다.

 

for i in range(N):
    for j in range(M):
        print(a[i][j] + b[i][j], end=' ')
    print()

for 문을 사용해 행의 수만큼 반복하는 문자열을 만든다.

for 문을 사용해 열의 수만큼 반복하는 문자열을 만든다.

 

print() 함수를 사용해 리스트 a에 i번째 리스트의 j번째 요소와 리스트 b의 i번째 리스트의 j번째 요소를 더한 값들을 출력한다.