글 읽기 - 7% 에서 틀렸습니다가 뜬 코드인데 무엇이 문제일까요?

girlera7 3년 전 0

7%에서 틀렸습니다가 뜬 코드인데 무엇이 문제일까요? 다른 분들 코드를 보고 비교해보고, 반례도 실행해 보았는데 오류를 못찾겠습니다.

from enum import Enum
from dataclasses import dataclass


@dataclass
class Point:
    y: int
    x: int


def is_valid_range(point, height, width):
    return 0 <= point.y < height \
           and 0 <= point.x < width


def is_boundary(point, height, width):
    return point.y == 0 or point.x == 0 or point.y == height - 1 or point.x == width - 1


def bfs(src: Point, fires: [Point]):
    dx = [-1, 1, 0, 0]
    dy = [0, 0, -1, 1]

    queue = []
    visited = [[False for _ in range(column)] for _ in range(row)]
    
    path = 0
    final_path = "IMPOSSIBLE"
    queue.append([src, path])
    visited[src.y][src.x] = True

    temp = []
    temp_fires = []
    while queue:
        temp = queue
        queue = []
        while temp:
            cur, cur_path = temp.pop(0)
            #print(cur, cur_path)
            if is_boundary(cur, row, column) and maze[cur.y][cur.x] != State.FIRED.value:
                final_path = cur_path + 1
                break

            for i in range(4):
                next = Point(cur.y + dy[i], cur.x + dx[i])
                if is_valid_range(next, row, column):
                    if not visited[next.y][next.x] and maze[cur.y][cur.x] != State.FIRED.value:
                        if maze[next.y][next.x] == State.EMPTY.value:
                            queue.append([next, cur_path + 1])
                        visited[next.y][next.x] = True
        if not queue:
            break
        temp_fires = fires
        fires = []
        while temp_fires:
            fire = temp_fires.pop(0)
            for i in range(4):
                next = Point(fire.y + dy[i], fire.x + dx[i])
                if is_valid_range(next, row, column):
                    if maze[next.y][next.x] == State.EMPTY.value or maze[next.y][next.x] == State.INITIAL.value:
                        maze[next.y][next.x] = State.FIRED.value
                        fires.append(next)
                        

    return final_path


class State(Enum):
    WALL = "#"
    EMPTY = "."
    INITIAL = "J"
    FIRED = "F"


row, column = map(int, input().split())
maze = []
fires = []
src = Point(-1, -1)

for r in range(row):
    m = input()
    lst = []
    for (c, elem) in enumerate(m):
        if elem == State.INITIAL.value:
            src = Point(r, c)
        elif elem == State.FIRED.value:
            fires.append(Point(r, c))
        lst.append(elem)
    maze.append(lst)

print(bfs(src, fires))

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

7% 에서 틀렸습니다가 뜬 코드인데 무엇이 문제일까요?

girlera7 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말