[백준] 2578. 빙고 python

🐳 링크

백준 2578번 빙고

🐕 문제

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

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

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

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

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

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

🐾 풀이 방법

시키는 대로 구현만 해서, 잘 푼 코드인진 모르겠다.
그냥 단순히 numbers 리스트 안의 number를 check라는 배열에서 동그라미친 후, (동그라미 치는 행위를 0->1로 구현함.) isBingo라는 함수를 통해 가로, 세로, 양 방향 대각선에서 빙고가 되는지 확인해주었다.

solution 코드

import sys
sys.stdin = open("input.txt","r")
def isBingo():

    res = 0
    # 가로 확인
    for i in range(5):
        flag = True
        if 0 in check[i]:
            flag = False
        if flag:
            res += 1

    # 세로 확인
    check_T = list(map(list,zip(*check)))
    for i in range(5):
        flag = True
        if 0 in check_T[i]:
            flag = False
        if flag:
            res += 1


    # 대각선
    r_flag, l_flag = True, True
    for i in range(5):
        if check[i][i] == 0:
            r_flag = False
        if check[i][4-i] == 0:
            l_flag = False
    if r_flag:
        res += 1
    if l_flag:
        res += 1

    if res >= 3:
        return True
    return False


bingo = list()
numbers = list()
check = [[0] * 5 for _ in range(5)]
count = 0

for _ in range(5):
    bingo.append(list(map(int, sys.stdin.readline().split())))
for _ in range(5):
    numbers.append(list(map(int, sys.stdin.readline().split())))

for number in numbers:
    while number:
        num = number.pop(0)
        count += 1

        for i in range(5):
            for j in range(5):
                if bingo[i][j] == num:
                    check[i][j] = 1
                    i = 5
                    break

        if count >= 5:
            if isBingo() == True:
                print(count)
                exit()