백준/ Gold 4 문제 , 백준 파이썬 12851, 숨바꼭질 2 [BFS]

백준/ Gold 4 문제 , 백준 파이썬 12851, 숨바꼭질 2 [BFS]

 

풀이 시간

 

 

Check Point !   ( 해당사항 ✓체크 )

1. 막힘 없이 수월하게 풀린 문제인가?

2. 1시간이내로 풀렸던 문제인가?

3. 1시간 이상 or 며칠을 두고 풀어봤더니 풀린 문제인가?

4. 시간을 써도 도무지 풀 수 없는 문제인가?

5. 솔루션을 찾아봤는가?

-------------------------------------------------------------------------------------------
난이도 체감
1. 최상
2. 상
3. 중
4. 하


<이해도>
1. 완벽히 이해
2. 다소 헷갈리는 부분들이 있음
3. 이해 못함


<덧붙일 말>
pypy3 로 먼저 통과한 뒤에 조금 더 코드를 수정해서 python3 로도 통과 성공

 

문제 출처

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

12851번: 숨바꼭질 2

 

 

 

풀이

 

 

첫 풀이 (pypy3만 통과)

 

 

처음에 store에 이미 들어가 있는지를 for문으로 확인해 주었는데 이 과정에서 시간이 많이 걸려 python3 에서는 시간 초과 발생

 

 

이후 아래 코드로 수정.

 

 

 

1. BFS

 

위에서는 for문으로 확인했지만 이 코드에선 visited를 2차원 배열로 선언하여 같은 이동횟수인지를 확인 해주고 같은 이동 횟수라면 store에 넣어주는 식으로 체크하였다.

 

이렇게 하니 훨씬 시간도 빨라지고 python3도 통과

 

 

 

2. 입력받기 및 BFS 호출&처리

 

 

N번 이동할때 이동될 수 있는 위치들을 모두 result에 넣어가며 bfs 를 호출한다.

 

bfs 를 한번 호출 할때마다 이동횟수가 하나씩 증가하는 것이므로 cnt 변수를 증가 시켜가며 bfs 를 호출할때 같이 넣어가며 체크해준다.

 

이때 N과 K가 같다면 0,0, 을 출력하는게 아닌 0,1 을 출력해야하는 것에 주의.

 

 

 

 

정답

 

 

from collections import deque

def bfs(l,k):
    c=0
    store=[]
    dq=deque()
    for i in l:
        dq.append(i)
    while dq:
        pos=dq.popleft()
        for i in (pos-1,pos+1,pos*2):
            if 0<=i<100001:
                if visited[i][0]==0:
                    store.append(i)
                    visited[i][0]=1
                    visited[i][1]=k
                elif visited[i][0]!=0 and visited[i][1]==k:
                    store.append(i)
                if i==K:
                    c+=1
    return store,c

N,K=map(int,input().split())
if N==K:
    print(0)
    print(1)
else:
    cnt=0
    result=[N]
    visited=[[0,0] for _ in range(100001)]
    while True:
        result,check=bfs(result,cnt)
        cnt+=1
        if check!=0:
            print(cnt)
            print(check)
            break