개발 연습장/백준 문제풀이

[파이썬, Python] 백준 2738: 행렬 덧셈

LooanCheong 2022. 12. 27. 17:29
반응형

문제

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

 

2738번: 행렬 덧셈

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

www.acmicpc.net

코드

n, m = map(int, input().split())
a, b = [], []

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

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

for row in range(n):
    for i in range(m):
        print(a[row][i] + b[row][i], end=' ')
    print()

설명

a와 b의 리스트를 만들어주어,
각각의 리스트 안에 또 다른 리스트를 만들어준다.(2차원 리스트)

a = [[1 1 1]            b = [[3 3 3
        [2 2 2]                   [4 4 4
        [0 1 0]]                  [5 5 100]]

각 원소의 개별 값에 차례대로 같은 위치의 값을 더해주어(1과 1, 2와 4, 0과 5, 1과 5, 1과 100)
계산한 값을 출력한다.

반응형