[Python] CodeTree - Trail 0. Chapter10 : 두 배열의 곱

 

 

 

1. 문제 설명

3행 3열의 배열 두 개가 주어지면 두 배열의 같은 위치에 있는 수의 곱을 출력하는 프로그램을 작성해보세요.

 

 

 

 

2. 입력 예시

3행 3열의 배열 두 개가 줄 간격을 사이에 두고 주어집니다.

1 2 3
4 5 6
7 8 9

2 3 4
5 6 7
8 9 10

 

 

 

 

3. 출력 예시

주어진 두 배열의 곱을 3행 3열의 배열로 입출력 예제와 같이 출력합니다.

2 6 12 
20 30 42
56 72 90

 

 

 

 

4. 문제 풀이

• 잘못된 풀이

A = [
    list(map(int,input().split()))
    for _ in range (3)
]

B = [
    list(map(int,input().split()))
    for _ in range (3)
]

for i in range(3):
    for j in range(3):
    	print(A[i][j]*B[i][j], end=" ")
    print()

처음엔 이렇게 작성했다. 이렇게 작성하면 배열 B에게 Index of range라는 오류가 발생하게 된다. 이상하다. A와 동일하게 작성해도 A는 오류 없이 배열 출력이 잘됐는데, 왜 B만 오류가 발생한걸까? 

입력할 때 제대로 입력되지 않았다고 추측했다. range(3)을 range(1)로 바꿔도 출력이 안되는걸 보니 배열 입력이 아예 안된게 분명했다. 그래서 다시 문제를 읽어봤다.

3행 3열의 배열 두 개가 줄 간격을 사이에 두고 주어집니다.

다시 읽어보니, 줄 간격을 사이에 두고 주어진다는 말이 걸렸다. 줄을 준다는 게 도대체 무슨 말일지 한참을 고민했고, 예제로 준 부분을 보니 줄 한 칸을 공백 처리 한 걸 확인할 수 있었다.

사용자가 한 번 엔터를 쳐서 공백을 주고 그 다음 리스트를 채운거였다. 그래서 공백을 준 부분이 B로 가고 그래서 범위를 좁혀 출력하려고 해도 아무것도 없으니 오류가 발생할 수 밖에 없었다.

 

 

• 제대로 된 풀이

A = [
    list(map(int,input().split()))
    for _ in range (3)
]

input()

B = [
    list(map(int,input().split()))
    for _ in range (3)
]

for i in range(3):
    for j in range(3):
    	print(A[i][j]*B[i][j], end=" ")
    print()

input( )을 넣어주니 깔끔하게 해결됐다. 해설에선 배열의 곱을 새로운 배열에 담았던데... 어차피 배열 모양으로 출력되는게 아닌 숫자로만 보이면 되는거라, 그냥 곱해주는 방식으로 풀었다.

 

 

• 해설

# 첫 번째 2차원 배열을 구현해 정수를 입력받습니다.
arr_1 = [
	list(map(int, input().split()))
	for _ in range(3)
]

input()

# 두 번째 2차원 배열을 구현해 정수를 입력받습니다.
arr_2 = [
	list(map(int, input().split()))
	for _ in range(3)
]

# 2차원 배열을 구현합니다.
arr_3 = [
    [0 for _ in range(3)]
    for _ in range(3)
]

# 두 배열의 곱을 새로운 배열에 담습니다.
for i in range(3):
	for j in range(3):
		arr_3[i][j] = arr_1[i][j] * arr_2[i][j]
	
# 새로운 배열을 출력합니다.
for i in range(3):        
    for j in range(3):    
        print(arr_3[i][j], end=" ")
    print()