[알고리즘 문제] Bingo (Python / 파이썬)

문제

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

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

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

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

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

입력

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

출력

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

---

문제 풀이 

빙고 문제는 사회자가 몇 번째 수를 부른 후 철수가 빙고를 외치는지 출력하는 문제이다.

따라서 사회자가 부르는 수를 차례대로 확인하며

철수의 빙고판에 똑같은 수를 0으로 만들어준 뒤 0으로 만들 때마다 

빙고인지 확인해주었다.

 

빙고는 행 혹은 열 혹은 대각선으로 모두 0이어야 1빙고가 성립된다. 

따라서 각 행, 열, 두 대각선을 탐색하며 빙고가 성립될 때마다 

빙고의 개수를 세어주었다. 그리고 구한 빙고의 개수를 반환해주었다. 

 

만약 구한 빙고의 개수가 3개 이상이라면 빙고를 외칠 수 있으므로 

사회자가 부른 수가 몇번 째 수인지 출력해주면 된다. 

---

My Code

# 빙고의 개수를 세어주는 함수 
def check():
    result = 0
    # 각 행에서 탐색 
    for i in range(5):
        count = 0
        for j in range(5):
            if myBingo[i][j] == 0: 
                count += 1
        if count == 5: # 행의 원소들이 모두 0이라면 개수 세어줌
            result += 1
    
    # 각 열에서 탐색 
    for j in range(5):
        count = 0
        for i in range(5):
            if myBingo[i][j] == 0:
                count += 1
        if count == 5: # 열의 원소들이 모두 0이라면 개수 세어줌
            result += 1
    
    # 오른쪽 아래로 내려오는 대각선 탐색
    count = 0
    for i in range(5):
        if myBingo[i][i] == 0:
            count += 1
    if count == 5: # 대각선 원소가 모두 0이라면 개수 세어줌
        result += 1

    # 왼쪽 아래로 내려오는 대각선 탐색
    count = 0
    for i in range(5):
        if myBingo[i][4 - i] == 0:
            count += 1
    if count == 5: # 대각선 원소가 모두 0이라면 개수 세어줌
        result += 1

    return result # 빙고의 개수를 반환

def simulate():
    count = 0
    result = 0
    for i in range(5):
        for j in range(5):
            x = bingoNum[i][j] # 사회자가 부른 수를 하나씩 확인
            count += 1         # 몇번 째 수인지 세어줌
            for a in range(5):
                for b in range(5):
                    if myBingo[a][b] == x: # 사회자가 부른 수라면
                        myBingo[a][b] = 0  # 지워줌
            result = check() # 지운 후 빙고가 몇 개인지 확인
            if result >= 3: # 3개 이상이면 빙고를 외칠 수 있으므로
                return count # 사회자가 부른 수가 몇번째 수인지 반환

# 철수의 빙고판 상태
myBingo = []
for _ in range(5):
    myBingo.append(list(map(int,input().split())))

# 사회자가 부른 수의 상태
bingoNum = []
for _ in range(5):
    bingoNum.append(list(map(int,input().split())))

print(simulate())