[Python/파이썬] 백준 2578번 빙고

2578번: 빙고

문제

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

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

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

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

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

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

입력

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

출력

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

코드

board = []
for _ in range(5):
    board.append(list(map(int, input().split())))
                 
def bingo():
    res = 0
    # 가로, 세로 확인
    for i in range(5):
        row_cnt = 0
        col_cnt = 0
        for j in range(5):
            if board[i][j] == 0:    # 가로 확인
                row_cnt += 1
            if board[j][i] == 0:    # 세로 확인
                col_cnt += 1
        if row_cnt == 5:
            res += 1
        if col_cnt == 5:
            res += 1
    # 대각선 확인
    diagnol1 = 0
    diagnol2 = 0
    for i in range(5):
        if board[i][i] == 0:    # 왼위->오른아래
            diagnol1 += 1
        if board[i][4-i] == 0:  # 오른위->왼아래
            diagnol2 += 1
    if diagnol1 == 5:
        res += 1
    if diagnol2 == 5:
        res += 1
    return res
    
    
for i in range(5):
    arr = list(map(int, input().split()))
    for j in range(5):
        # 붙려진 번호 지우기
        for r in range(5):
            for c in range(5):
                if board[r][c] == arr[j]:
                    board[r][c] = 0
        # 빙고 됐는지 확인
        if bingo() >= 3:
            print(i*5+j+1)
            exit()