[백준] 브론즈5_2738_행렬 덧셈 (python)

juyeon·2022년 7월 1일
0

코딩테스트(python)

목록 보기
16/22

백준

행렬 덧셈

문제

: n, m 주어짐.
n줄까지의 행렬과 그 이후 또 n줄만큼의 행렬 각 원소를 더한 수를 출력하기.

나의 풀이

n, m = map(int, input().split())
matrix = [list(map(int, input().split())) for _ in range(2 * n)]
m_a = matrix[:n] #리스트 슬라이싱
m_b = matrix[n:]
c = [[0] * m for _ in range(n)] #빈 행렬 리스트 생성

for a in range(n):
    for b in range(m):
        c[a][b] = m_a[a][b] + m_b[a][b] #빈 리스트 채우기
        
for i in range(n):
    for j in range(m):
        print(c[i][j], end = ' ') #스페이스바로 구분하여 출력
    print() #행 구분을 위해 행 끝날 때마다 개행 문자 출력
profile
내 인생의 주연

0개의 댓글