강의

멘토링

커뮤니티

인프런 커뮤니티 질문&답변

이헌기님의 프로필 이미지
이헌기

작성한 질문수

파이썬 알고리즘 문제풀이 입문(코딩테스트 대비)

14. 안전영역(DFS)

BFS방식으로 풀었는데 테케 2번부터 타임리밋뜹니다...

작성

·

155

0

도와주세요... 

from collections import deque

def BFS(h):
    while(dq):
        t = dq.popleft()
        chk[t[0]][t[1]] = 1

        for i in range(4):
            xx = t[0] + dx[i]
            yy = t[1] + dy[i]
            if(0 <= xx < N and 0 <= yy < N and board[xx][yy] > h and chk[xx][yy] == 0):
                dq.append((xx, yy))
    return

if __name__ == '__main__':
    dx = [-1, 0, 1, 0]
    dy = [0, 1, 0, -1]
    N = int(input())
    ans = 0
    board = [list(map(int, input().split())) for _ in range(N)]
    chk = list([0] * N for _ in range(N))
    dq = deque([])

    for h in range(100):
        cnt = 0
        chk = list([0] * N for _ in range(N))
        for i in range(N):
            for j in range(N):
                if(chk[i][j] == 0 and board[i][j] > h):
                    cnt += 1
                    dq.append((i, j))
                    BFS(h)
        ans = max(ans, cnt)
        if(cnt == 0):
            break
    print(ans)

답변 1

1

김태원님의 프로필 이미지
김태원
지식공유자

안녕하세요^^

BFS방식일 뿐 영상의 DFS와 동일한 코드나 다름없습니다. 제가 어떻게 해줄 수 있는 것 같지는 않습니다.

이문제 처럼 퍼저나가며 확인하는 블러드필 형태의 문제는 파이썬에서는 재귀를 이용한 DFS방식을 쓰시고 sys.setrecursionlimit(10**6) 이 코드를 넣어 주는게 제일 좋은 방법인 것 같습니다.

링코된 페이지는 백준에서 파이썬으로 BFS해서 통과한 코드라고 하니 확인해보세요.

https://it-garden.tistory.com/181

이헌기님의 프로필 이미지
이헌기

작성한 질문수

질문하기