[baekjoon] 백준 2578번(파이썬): 빙고

문제

2578번: 빙고

알고리즘

- 반복문을 통해 사회자가 외친 빙고 번호를 0으로 초기화

- 가로, 세로, 대각선이 빙고인지 확인 후 빙고이면 카운트

- 카운트한 빙고가 3개 이상이라면 빙고를 부른 횟수를 출력 후 시스템 종료

코드

import sys


def solve(target, graph):
    for i in range(5):
        for j in range(5):
            if target == graph[i][j]:
                graph[i][j] = 0
                return


def check(graph):
    cnt = 0
    for i in range(5):
        temp = 0
        for j in range(5):
            if graph[i][j] == 0:
                temp += 1
        if temp == 5:
            cnt += 1
    for i in range(5):
        temp = 0
        for j in range(5):
            if graph[j][i] == 0:
                temp += 1
        if temp == 5:
            cnt += 1

    if graph[0][4] == graph[1][3] == graph[2][2] == graph[3][1] == graph[4][0] == 0:
        cnt += 1
    if graph[0][0] == graph[1][1] == graph[2][2] == graph[3][3] == graph[4][4] == 0:
        cnt += 1

    return cnt


bingo = [list(map(int, sys.stdin.readline().split())) for _ in range(5)]
answer_bingo = [list(map(int, sys.stdin.readline().split())) for _ in range(5)]


answer = 0
for i in range(5):
    for j in range(5):
        answer += 1
        solve(answer_bingo[i][j], bingo)
        if check(bingo) >= 3:
            print(answer)
            exit()

github

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법(Python)