2738 행렬 덧셈 (python)

풀이

n, m = map(int, input().split())

 

A,B = [],[]

 

# A 행렬 입력받기

for i in range(n):

    a = list(map(int,input().split()))

    A.append(a)

# B 행렬 입력받기

for i in range(n):

    b = list(map(int,input().split()))

    B.append(b)

 

 # 더한 후 출력

for i in range(n): # 행 'i'

    for j in range(m): # 열 'j'

        # 각 행('i')dhk 각 열('j')에 대해 더한 값을 출력(이차원 배열)

        result = A[i][j] + B[i][j]

        print(result, end=' ')

    print()

 

이차원 배열로 더하는 이유 (더해지는 과정)

 

입력 예시 (n = 3 (행의 수), m = 3 (열의 수))

3 3
1 1 1
2 2 2
0 1 0
3 3 3
4 4 4
5 5 100

 

첫 번째 루프 (i = 0)

• j = 0: result = A[0][0] + B[0][0] = 1 + 3 = 4
• j = 1: result = A[0][1] + B[0][1] = 1 + 3 = 4
• j = 2: result = A[0][2] + B[0][2] = 1 + 3 = 4

이후 출력: 4 4 4

두 번째 루프 (i = 1)

• j = 0: result = A[1][0] + B[1][0] = 2 + 4 = 6
• j = 1: result = A[1][1] + B[1][1] = 2 + 4 = 6
• j = 2: result = A[1][2] + B[1][2] = 2 + 4 = 6

이후 출력: 6 6 6

세 번째 루프 (i = 2)

• j = 0: result = A[2][0] + B[2][0] = 0 + 5 = 5
• j = 1: result = A[2][1] + B[2][1] = 1 + 5 = 6
• j = 2: result = A[2][2] + B[2][2] = 0 + 100 = 100

이후 출력: 6 6 6

 

최종 출력:

4 4 4
6 6 6
5 6 100