백준 단계별로 풀어보기/2차원 배열

백준 2738번 행렬 덧셈(Python / 파이썬)

coding232624 2023. 3. 25. 22:39

주소

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

 

2738번: 행렬 덧셈

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

www.acmicpc.net

 

문제

N, M을 입력받은 후 N*M행렬을 입력해 주고 그 후에 입력한 2개의 행렬의 합을 구하는 문제

 

코드 / 해설

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
N,M = map(int, input().split())
 
A,B = [],[]
 
 
for i in range(N):
    a = list(map(int,input().split()))
    A.append(a)
        
for i in range(N):
    b = list(map(int,input().split()))
    B.append(b)
 
for i in range(N):
    for j in range(M):
        result = A[i][j] + B[i][j]
        print(result,end=' ')
    print()
        
 
cs

=> 2차원 행렬에 대해 이해하고 있고 .append를 통해 리스트 자체를 입력하는 법을 알고 있다면 어렵지 않은 문제이다. 리스트 자체를 저장하게 될 경우 차원이 늘어난다는 점을 기억하면 좋을듯한 문제이다