본문 바로가기
코딩테스트

[파이썬] 백준 2578 빙고

by 왕자두 2023. 6. 10.

[파이썬] 백준 2578번 빙고

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

 

2578번: 빙고

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

www.acmicpc.net

 

[문제 이해하기]

문제를 이해하는 것 자체는 크게 어렵지 않았던 것 같다. 우리가 아는 빙고 게임과 동일한 방식으로 문제를 풀면 되는데 5x5 빙고판에 3 빙고가 될 때까지 부른 숫자의 개수를 출력하면 되는 문제였다.

여기서 내가 헷갈렸던 포인트는

1. 2차원 배열 두 개를 만들어서 비교해야하는가 or 2차원 배열&1차원 배열 하나씩 만들어서 비교를 해야하는가

 => 풀다보니 철수가 만든 빙고판은 2차원 배열, 사회자가 부른 숫자들은 1차원 배열로 해야 for문을 세 번만 돌 수 있다는 사실을 깨달았다. 만약 전자를 선택했다면 for문만 네 번 돌 뻔 했다.

2. 어떻게 해야 사회자가 부른 값을 표시할 수 있을까?

=> 새로운 배열을 또 만들어서 관리해야하나 생각했는데 굳이 그럴 필요 없이 1~25가 아닌 숫자로 그 자리를 바꿔주면 될 것 같다고 생각했다. 그래서 제일 무난한 0으로 바꿔주기로 하였다.

3. 가로, 세로는 어떻게 확인을 해야할까?

=> 가로는 2차원 배열이니까 한 행씩 비교하면 될 것 같다고 생각했다. 하나하나 0으로 찾을 수 있지만 찾아보니 배열에서 원하는 int 값을 찾을 수 있는 함수인 `count` 함수가 있어서 count 함수를 사용하여 0이 5개일 경우에 대한 처리를 해주었다. 세로는 그냥 하나하나 비교했다.

 

[문제 풀이 중 여러 번 틀린 부분]

  • 가로, 세로, 왼쪽 대각선, 오른쪽 대각선을 비교할 때 각각 변수를 초기화 해줘야되는데 이 부분을 틀렸었고 특히 세로 비교를 할 때 0의 개수를 세는 변수를 for문 안쪽에 넣어서 한 줄마다 초기화 했어야 하는데 이걸 몰라서 오랫동안 붙잡고 있었다.
  • 사회자가 부르는 숫자가 일정 개수 이상 되어야 그때부터 3 빙고가 완성될 수 있는 조건을 만족시키게 되는데 그게 몇인지 찾는 게 조금 어려웠다. 처음에는 겹치는 숫자에 대해서 생각하지 못해서 15라고 했다가 최대 2개까지라고 생각해서 13이라고 했다가 최대 3개의 숫자까지 겹칠 수 있다는 걸 알았다. 삼각형으로 가장 위쪽 행, 가장 오른쪽 열, 그리고 왼쪽 대각선 이렇게 되면 0행 0열, 0행 4열과 4행 4열 값이 겹치게 된다. 그래서 12로 했더니 맞았다.
  • for문을 마지막까지 계속 돌리는데 만약 3 빙고가 되면 멈출 수 있도록 break도 추가해야한다.

 

[최종 코드]

import sys

input = sys.stdin.readline

n = [list(map(int, input().split())) for _ in range(5)]
m = []

for i in range(5):
    m += list(map(int, input().split()))

#하나씩 돌면서 확인
def check(n):
    check_num = 0

    #가로
    for i in n:
        if i.count(0) == 5:
            check_num+=1

    #세로
    for i in range(5):
        num = 0
        for j in range(5):
            if n[j][i] == 0:
                num +=1
        if num == 5:
            check_num+=1

    #왼쪽 대각선
    num = 0
    for i in range(5):
        if n[i][i] == 0:
            num+=1
    if num == 5:
        check_num+=1

    #오른쪽 대각선
    num = 0
    for i in range(5):
        if n[i][4-i] == 0:
            num+=1
    if num == 5:
        check_num+=1

    return check_num

count = 0

for i in range(25):
    for j in range(5):
        for k in range(5):
            if m[i] == n[j][k]:
                n[j][k] = 0
                count+=1

    if count >= 12:
        if check(n) >= 3:
            print(count)
            break