백준 / 2578번 빙고 - 구현 / Python 파이썬

 
*문제 출처는 백준에 있습니다.

문제 제목: 2578번 빙고

문제 사이트: https://www.acmicpc.net/problem/2578
 
 

---

나의 풀이
 
후 빡코딩이였다 오랜만에 집중함
for문이 반복되는만큼 탈출해야하는 것이랑 제대로 다 생각해서 코딩해야함
maps 같이 2차원 행렬 사용된다싶으면 무조건  pseudo  code 만들고 진입하자 ㅠㅠ
 

import sys

input = sys.stdin.readline
# 세 개 이상 그어지는 순간 "빙고"
maps = []
visited = [[False] * 5 for i in range(5)]
cnt = 0
answer = 0 # 사회자가 몇번 불렀는지 카운트
# 체크했던 인덱스들 저장
c_row = [0] * 5
c_cal = [0] * 5

c_right = False
c_left = False
for i in range(5):
    maps.append(list(map(int,input().split())))

def check_row():
    global visited, cnt, answer
    global c_row
    if cnt == 3:
        return
    
    for i in range(5):
        if c_row[i] != 1:
            if visited[i][:] == [True, True, True, True, True]:
                cnt += 1
                c_row[i] = 1
        else:
            continue


def check_cal():
    global visited, cnt, answer
    global c_cal
    if cnt == 3:
        return
    
    for i in range(5):
        tmp = []
        if c_cal[i] == 1:
            continue
        tmp_num = 0
        for j in range(5):
            tmp_num = i
            if visited[j][i] == True:
                tmp.append(True)

        if tmp == [True, True, True, True, True]:
            cnt += 1
            c_cal[tmp_num] = 1

def check_right(): # 오른쪽 위부터 시작하는 대각선
    global visited, cnt, answer
    global c_right
    tmp = []
    if cnt == 3:
        return
    
    if c_right == True:
        return
    else:
        for i in range(5):
            if visited[i][5-i-1] == True:
                tmp.append(True)
        if tmp == [True, True, True, True, True]:
            cnt += 1
            c_right = True

def check_left(): # 왼쪽 위부터 시작하는 대각선
    global visited, cnt, answer
    global c_left
    tmp = []
    if cnt == 3:
        return
    
    if c_left == True:
        return
    else:
        for i in range(5):
            if visited[i][i] == True:
                tmp.append(True)
        if tmp == [True, True, True, True, True]:
            cnt += 1
            c_left = True

def call_number(number):
    global answer
    answer += 1
    for i in range(5):
        for j in range(5):
            if maps[i][j] == number:
                visited[i][j] = True
                check_row()
                check_cal()
                check_right()
                check_left()
                return


while cnt < 3:
    numbers = list(map(int,input().split()))
    for i in range(5):
        call_number(numbers[i])
        if cnt >= 3:
            break
print(answer)

 

GPT 답안
훨씬 최적화 잘돼있음 자주 보고 익히자

import sys

input = sys.stdin.readline
# 세 개 이상 그어지는 순간 "빙고"
maps = []
visited = [[False] * 5 for _ in range(5)]
cnt = 0
answer = 0  # 사회자가 몇 번 불렀는지 카운트

# 체크했던 인덱스들 저장
c_row = [0] * 5
c_cal = [0] * 5
c_right = False
c_left = False

for i in range(5):
    maps.append(list(map(int, input().split())))

def check_all():
    global c_row, c_cal, c_left, c_right
    bingo = 0

    # 가로 줄 검사
    for i in range(5):
        if c_row[i] == 0 and visited[i] == [True] * 5:
            c_row[i] = 1
            bingo += 1

    # 세로 줄 검사
    for j in range(5):
        if c_cal[j] == 0 and all(visited[i][j] for i in range(5)):
            c_cal[j] = 1
            bingo += 1

    # 왼쪽 대각선
    if not c_left and all(visited[i][i] for i in range(5)):
        c_left = True
        bingo += 1

    # 오른쪽 대각선
    if not c_right and all(visited[i][4 - i] for i in range(5)):
        c_right = True
        bingo += 1

    return bingo

def call_number(number):
    global answer, cnt
    answer += 1
    for i in range(5):
        for j in range(5):
            if maps[i][j] == number:
                visited[i][j] = True
                cnt += check_all()
                return

while cnt < 3:
    numbers = list(map(int, input().split()))
    for i in range(5):
        call_number(numbers[i])
        if cnt >= 3:
            break

print(answer)

---

※ 알아야 할 것

- all()은 파이썬에서 자주 쓰이는 내장 함수로, 모든 요소가 참(True)이면 True, 하나라도 거짓(False)이면 False를 반환