https://www.acmicpc.net/problem/2738
문제
N*M크기의 두 행렬 A와 B가 주어졌을 때, 두 행렬을 더하는 프로그램을 작성하시오.
풀이
행렬의 크기를 정해주고 append 시킨다음 인덱스 번호를 잘 지정해서 반복문으로 더해주면 된다.
코드
# MxN 행렬의 크기 입력 받기
M, N = map(int, input().split())
A, B = [], []
# 행렬 A 입력 받기
for _ in range(M):
row = list(map(int, input().split()))
A.append(row)
# 행렬 B 입력 받기
for _ in range(M):
row = list(map(int, input().split()))
B.append(row)
# 두 행렬의 합 구하기
for row in range(M):
for col in range(N):
print(A[row][col] + B[row][col], end=' ')
print() # 열 바꿈
'백준 > python' 카테고리의 다른 글
2751:수 정렬하기 2(python) (0) | 2023.09.14 |
---|---|
10798:세로읽기(python) (0) | 2023.09.14 |
2998:8진수(python) (0) | 2023.09.14 |
1259:팰린드롬 수(python) (0) | 2023.09.14 |
2693:N번째 큰수(python) (0) | 2023.09.14 |