백준 2578번 - 빙고 (파이썬/Python)

문제

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

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

 

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

입력

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

출력

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

 

문제 링크

https://www.acmicpc.net/problem/2578

2578번: 빙고

답안

c = [list(map(int, input().split())) for _ in range(5)]
mc = []
for _ in range(5):
    mc += list(map(int, input().split()))


def check():
    bingo = 0

    # 가로 확인
    for x in c: 
        if x.count(0) == 5:
            bingo += 1

    # 세로 확인
    for i in range(5): 
        y = 0
        for j in range(5):
            if c[j][i] == 0:
                y += 1
        
        if y == 5:
            bingo += 1

    # 왼쪽위부터 시작하는 대각선 확인
    d1 = 0
    for i in range(5):
        if c[i][i] == 0:
            d1 += 1
    
    if d1 == 5:
        bingo += 1

    # 오른쪽위부터 시작하는 대각선 확인
    d2 = 0
    for i in range(5):
        if c[i][4-i] == 0:
            d2 += 1
    
    if d2 == 5:
        bingo += 1

    return bingo 


cnt = 0
for i in range(25):
    for x in range(5):
        for y in range(5):
            if mc[i] == c[x][y]:
                c[x][y] = 0
                cnt += 1

    if cnt >= 12:
        result = check()

        if result >= 3:
            print(i+1)
            break

후기

빙고 3개가 충족되기 위한 최소 횟수(12회) 이후 부터 빙고인지를 확인하는 check함수를 호출해준다.

만약 결과가 3 이상이면 (2에서 4로 넘어가는 경우가 있을 수 있기 때문에 3 이상으로 잡는다.) i+1값을 호출한다!

뭔가 머리쓰는 느낌의 문제라 재미있었다,,^^