[백준🥈4] 14499 - 주사위 굴리기 (Python)

Review

[1차 코드 문제점]

1. 대각선 체크 인덱스 오류:
대각선 체크 함수에서 인덱스가 잘못되었습니다.
check_dia1_list와 check_dia2_list의 좌표가 잘못되었습니다.

2. 함수에서 cnt 증가 시점 및 리턴 오류:
각 체크 함수에서 cnt를 증가시키고 True를 반환하는 부분이 올바르지 않습니다.
return 문이 반복문 안에 있어서 한 번만 체크하고 종료되는 문제가 있습니다.

3. 중복 체크 방지:
한 번 체크된 빙고 라인이 여러 번 카운트되지 않도록 개선해야 합니다.

4. 중앙 값부터 시작:
빙고 판의 중앙부터 값을 체크해야 합니다.

✨ HOW? 처음 아이디어

가로, 세로, 대각선 확인하는 함수 만들고
나의 빙고판을 이중 for문으로 돌면서 사회자가 부르는 숫자를 queue로 두고 한 개씩 꺼내면서 비교

사회자의 수를 나의 빙고판에서 체크할 때 마다, 해당 수는 0으로 바꿔줘고 for문 나가기..

1차 Code - 실패

from collections import deque

binggo = []
for _ in range(5):
    binggo.append(list(map(int, input().split())))

check= deque()
for _ in range(5):
    check_list = list(map(int, input().split()))
    for c in check_list:
        check.append(c)

cnt = 0

def check_row():
    global cnt
    
    for i in range(5):
        binggo_check = True
        for j in range(5):
            if binggo[i][j] != 0:
                binggo_check = False
        if binggo_check:
            cnt += 1
            return True
    
def check_col():
    global cnt
    for i in range(5):
        binggo_check = True
        for j in range(5):
            if binggo[j][i] != 0:
                binggo_check = False
                return
        if binggo_check:
            cnt += 1
        return True
    
def check_dia1():
    global cnt
    
    check_dia1_list = [(1,1),(2,2),(3,3),(4,4),(5,5)]
    binggo_check = True
    for x, y in check_dia1_list:
        if binggo[x][y] != 0:
            binggo_check = False
            return False
    if binggo_check:
        ans += 1
        return True

def check_dia2():
    global cnt
    check_dia2_list = [(4,0),(3,1),(2,2),(1,3),(0,4)]
    binggo_check = True
    for x, y in check_dia2_list:
        if binggo[x][y] != 0:
            binggo_check = False
            return False
    if binggo_check:
        cnt += 1
        return True

ans = 0
while check:
    go = check.popleft()
    flag = False

    for i in range(5):
        for j in range(5):      
            if binggo[i][j] == go:
                print("binggo[i][j]", binggo[i][j])
                binggo[i][j] = 0
                flag = True
                ans += 1
                break
            
        if flag:
            break
        
    check_row()
    print("check_row", cnt)
    if cnt == 3:
        print(ans)
        break
    
    check_col()
    print("check_col", cnt)
    if cnt == 3:
            print(ans)
            break
        
    check_dia1()
    print("check_dia1", cnt) 
    if cnt == 3:
            print(ans)
            break
        
    check_dia2()
    print("check_dia2", cnt)
    if cnt == 3:
            print(ans)
            break
        

2차 Code - 정답

from collections import deque
binggo = []
for _ in range(5):
    binggo.append(list(map(int, input().split())))

check = deque()
for _ in range(5):
    check_list = list(map(int, input().split()))
    for c in check_list:
        check.append(c)

def check_row():
    global cnt
    for i in range(5):
        if all(binggo[i][j] == 0 for j in range(5)):
            cnt += 1

def check_col():
    global cnt
    for i in range(5):
        if all(binggo[j][i] == 0 for j in range(5)):
            cnt += 1

def check_dia1():
    global cnt
    if all(binggo[i][i] == 0 for i in range(5)):
        cnt += 1

def check_dia2():
    global cnt
    if all(binggo[i][4-i] == 0 for i in range(5)):
        cnt += 1

ans = 0
while check:
    go = check.popleft()
    
    for i in range(5):
        for j in range(5):
            if binggo[i][j] == go:
                binggo[i][j] = 0
                break
        else:
            continue
        break
    
    cnt = 0
    check_row()
    check_col()
    check_dia1()
    check_dia2()
    
    if cnt >= 3:
        print(ans + 1)
        break
    
    ans += 1

3차 Code - 정답+개선

from collections import deque

# 빙고판 입력
binggo = []
for _ in range(5):
    binggo.append(list(map(int, input().split())))

# 사회자가 부르는 숫자 순서 입력
check = deque()
for _ in range(5):
    check_list = list(map(int, input().split()))
    for c in check_list:
        check.append(c)

def check_bingo():
    cnt = 0
    # 행 검사
    for i in range(5):
        if all(binggo[i][j] == 0 for j in range(5)):
            cnt += 1
    # 열 검사
    for i in range(5):
        if all(binggo[j][i] == 0 for j in range(5)):
            cnt += 1
    # 대각선 검사 (왼쪽 위에서 오른쪽 아래)
    if all(binggo[i][i] == 0 for i in range(5)):
        cnt += 1
    # 대각선 검사 (오른쪽 위에서 왼쪽 아래)
    if all(binggo[i][4-i] == 0 for i in range(5)):
        cnt += 1
    return cnt

ans = 0
while check:
    go = check.popleft()
    
    found = False

    for i in range(5):
        if found:
            break
        for j in range(5):
            if binggo[i][j] == go:
                binggo[i][j] = 0
                found = True
                break
    
    if check_bingo() >= 3:
        print(ans + 1)
        break
    
    ans += 1

시간복잡도

O(n)