빙고 코딩테스트 연습풀이 [백준 2578/파이썬/구현]

문제 출처: https://www.acmicpc.net/problem/2578

2578번: 빙고

 

문제

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

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

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

 

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

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

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

 

입력

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

 

출력

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

 

 

정답

arr = [list(map(int, input().split())) for _ in range(10)]
cnt = 0

for i in range(5):
    for j in range(5):
        tmp = arr[i+5][j]
        line = 0
        cnt+=1
        for x in range(5):
            for y in range(5):
                if arr[x][y] == tmp:
                    arr[x][y] = 0
                    # 대각선 빙고일 경우
                    if arr[0][0] == arr[1][1] == arr[2][2] == arr[3][3] == arr[4][4]:
                        line+=1
                    if arr[0][4] == arr[1][3] == arr[2][2] == arr[3][1] == arr[4][0]:
                        line+=1

                    tmp_sum = [0,0,0,0,0]
                    for a in range(5):
                        #가로 빙고일 경우
                        if sum(arr[a]) == 0:
                            line+=1
                            if line == 3:
                                print(cnt)
                                exit()
                        
                        for b in range(5):
                            #세로 빙고를 위한 세팅
                            tmp_sum[b]+=arr[a][b]
                    line+=tmp_sum.count(0)
                    if line >=3:
                        print(cnt)
                        exit()

다중 for문을 사용해서 푸는 방법을 사용했습니다.

빙고가 5x5인 경우이기 때문에 시간복잡도는 크게 생각하지 않고 풀었습니다.

조건이 더 있을 경우에는 함수를 사용해서 푸세요.