[Python] 백준(BOJ) 2578번: 빙고

문제

2578번: 빙고

풀이

가로세로를 확인하는 함수, 대각선 확인하는 함수를 만든 후 인풋에 따라 기본 빙고판을 채웠고, 

사회자가 부르는 순서대로 하나씩 지워가면서 빙고인지 확인해줬다. 

import sys

def check_line():
    bingo = 0
    for i in range(5):
        bool1 = True
        bool2 = True
        for j in range(5):
            if board[j][i] != -1:
                bool1 = False
            if board[i][j] != -1:
                bool2 = False

        if bool1:
            bingo += 1
        if bool2:
            bingo += 1
    return bingo


def check_diagonal():
    bingo = 0
    bool1 = True
    bool2 = True
    for i in range(5):
        if board[i][i] != -1:
            bool1 = False
        if board[i][4-i] != -1:
            bool2 = False
    if bool1:
        bingo += 1
    if bool2:
        bingo += 1

    return bingo


dic = dict()
board = []
# 빙고판 만들기
for i in range(5):
    li = list(map(int, sys.stdin.readline().strip().split(" ")))
    board.append(li)
    for j in range(5):
        dic[li[j]] = (i, j)


count = 0
flag = False
for _ in range(5):
    li = list(map(int, sys.stdin.readline().strip().split(" ")))
    for each in li:
        count += 1
        r, c = dic[each]
        board[r][c] = -1

        if (check_diagonal() + check_line()) >= 3:
            print(count)
            flag = True
            break
    if flag:
        break