[백준] 2578번 빙고 [Python]

5x5 빙고판$($1~25$)$에서 사회자가 불러주는 숫자를 지워가며 3 줄을 완성 시키는 게임

문제

빙고 게임은 다음과 같은 방식으로 이루어진다.

먼저 아래와 같이 25개의 칸으로 이루어진 빙고판에 1부터 25까지 자연수를 한 칸에 하나씩 쓴다

 

다음은 사회자가 부르는 수를 차례로 지워나간다. 예를 들어 5, 10, 7이 불렸다면 이 세 수를 지운 뒤 빙고판의 모습은 다음과 같다.

 

차례로 수를 지워가다가 같은 가로줄, 세로줄 또는 대각선 위에 있는 5개의 모든 수가 지워지는 경우 그 줄에 선을 긋는다.

 

이러한 선이 세 개 이상 그어지는 순간 "빙고"라고 외치는데, 가장 먼저 외치는 사람이 게임의 승자가 된다.

 

철수는 친구들과 빙고 게임을 하고 있다. 철수가 빙고판에 쓴 수들과 사회자가 부르는 수의 순서가 주어질 때, 사회자가 몇 번째 수를 부른 후 철수가 "빙고"를 외치게 되는지를 출력하는 프로그램을 작성하시오.

입력

첫째 줄부터 다섯째 줄까지 빙고판에 쓰여진 수가 가장 위 가로줄부터 차례대로 한 줄에 다섯 개씩 빈 칸을 사이에 두고 주어진다. 여섯째 줄부터 열째 줄까지 사회자가 부르는 수가 차례대로 한 줄에 다섯 개씩 빈 칸을 사이에 두고 주어진다. 빙고판에 쓰여진 수와 사회자가 부르는 수는 각각 1부터 25까지의 수가 한 번씩 사용된다.

출력

첫째 줄에 사회자가 몇 번째 수를 부른 후 철수가 "빙고"를 외치게 되는지 출력한다.

 

import sys
input = sys.stdin.readline
A = [list(map(int, input().split())) for _ in range(5)]  # 빙고판
B = [list(map(int, input().split())) for _ in range(5)]  # 사회자가 부르는 번호
bingo = 0

def f(A, i, j): 					 # 대각선, 가로, 세로 확인
    count = 0
    if i == j and A[0][0] + A[1][1] + A[2][2] + A[3][3]+ A[4][4] == 0:
        count += 1
    if i + j == 4 and A[0][4] + A[1][3] + A[2][2] + A[3][1]+ A[4][0] == 0:
        count += 1
    if sum(A[i]) == 0:
        count += 1
    if A[0][j] + A[1][j] + A[2][j] + A[3][j]+ A[4][j] == 0:
        count += 1
    return count

for r in range(5):
    for c in range(5):
        n = B[r][c]
        for i in range(5):
            if n in A[i]:
                for j in range(5):
                    if A[i][j] == n:		# 사회자가 불러준 숫자 지우기
                        A[i][j] = 0
                        break
                bingo += f(A, i, j)		# 지운 좌표에서 줄을 그을 수 있는지 확인하기
                if bingo >= 3:	# 3 줄 이상이면
                    print(r * 5 + c + 1)	# 답 출력
                    exit()			# 종료

 

반복문 종료
$return(), break(), exit()$

1. $return()$:
   - return 키워드는 함수 내에서 사용되며, 함수를 호출한 곳으로 값을 반환하는 역할을 합니다.
   - 함수가 실행 중 return 문을 만나면 해당 함수의 실행은 종료되고, 반환값이 함수를 호출한 곳으로 전달됩니다.
   - 예를 들어, 아래의 함수 `add`는 두 개의 숫자를 받아서 그 합을 반환하는 함수입니다.

def add(a, b):
   return a + b

result = add(3, 5)  # result 변수에는 8이 저장됨

2. $break()$:
   - break 키워드는 반복문$($for 또는 while$)$ 내에서 사용되며, 반복문을 종료하는 역할을 합니다.
   - 일반적으로 조건문과 함께 사용되어 특정 조건을 만족할 때 반복문을 빠져나오는데 사용됩니다.
   - 예를 들어, 아래의 반복문은 리스트에서 숫자 5를 찾으면 반복문을 종료합니다.

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

for num in numbers:
   if num == 5:
       print("숫자 5를 찾았습니다!")
       break

3. $exit()$:
   - exit() 함수는 파이썬 프로그램을 강제로 종료하는 역할을 합니다.
   - exit()를 호출하면 프로그램이 즉시 종료되며, 나머지 코드는 실행되지 않습니다.
   - 주로 특정 조건에 따라 프로그램을 강제로 종료해야 할 때 사용됩니다.
   - 주의: exit() 함수는 프로그램의 실행 도중에 사용하면 모든 작업을 종료하므로 신중하게 사용해야 합니다.

   answer = input("프로그램을 종료하시겠습니까? (yes/no): ")
   if answer == 'yes':
       print("프로그램을 종료합니다.")
       exit()
   else:
       print("프로그램을 계속 실행합니다.")

 

break는 하나의 반복문에서 빠져 나올 수 있기 때문에 다중 반복문 내에서는 exit, 함수 내라면 return을 사용할 수 있다.
다중 반복문 중 일부에서만 빠져 나오고 싶다면 번거롭겠지만, 아래와 같이 if문을 활용해 break를 여러 번 사용할 수도 있다. / While문을 잘 활용하는 것도 좋은 방법이다.

s = 0
c = 0
for j in range(100)
	for i in range(j, 100):
        s += 3000
        if s > 1234:
            c = 1
            break
	if c == 1:
    break

 

예제 입력 1

11 12 2 24 10
16 1 13 3 25
6 20 5 21 17
19 4 8 14 9
22 15 7 23 18
5 10 7 16 2
4 22 8 17 13
3 18 1 6 25
12 19 23 14 21
11 24 9 20 15

예제 출력 1 

15