20-alstjr7437

[alstjr7437]

🔗 문제 링크

숨바꼭질

✔️ 소요된 시간

30분

📜 문제 설명

1. n과 k를 입력받습니다.
2. 숫자 n은 -1, +1, *2가 가능합니다.
3. n에서 k까지 가는데 몇번 걸리는지 출력합니다.

예제) n -> 5 k -> 17
n : 5 -> 10 -> 9 -> 18 -> 17

✨ 수도 코드

오래만에 백준 아래에 있는 문제 유형을 안보고 풀었는데
처음에 사실 아침 알고리즘 수업들으며 문제 선정하면서 오잉 DP 문제 아닌가 생각해놓고

다른 공부를 하면서 틈이 날때(길 걸을때 등등..) 생각해보니까 트리를 만들면 되는 문제였습니다!!

일단 심심할때 끄적인걸로는 아래 사진과 같이

-1, +1, *2로 3갈래로 나눠지면서
만약에 k라는 부분에 오게 되면
해당 부분의 Level을 출력하면 되는 문제였구나 하면서 풀었습니다!!

그리고 무한 루프 및 시간을 줄이기 위해 중복으로 방문하는 숫자는 Level로 visited 처리 해줬습니다.

💻 최종 코드

from collections import deque

n, k = map(int, input().split())

result = [0] * 100001
result[n] = 1

queue = deque()
queue.append(n)

while queue:
    now = queue.popleft()
    if now == k:
        break
    # -1 로직
    if now - 1 >= 0 and result[now-1] == 0 :
        queue.append(now-1)
        result[now-1] = result[now] + 1
    # +1 로직
    if now + 1 < len(result) and result[now+1] == 0:
        queue.append(now+1)
        result[now+1] = result[now] + 1
    # *2 로직
    if now * 2 < len(result) and result[now*2] == 0:
        queue.append(now*2)
        result[now*2] = result[now] + 1

print(result[k] - 1)

📚 새롭게 알게된 내용

문제를 보고 이제 그래프를 통해 구현해야 하는지 조금씩 감을 잡아가고 있는 것 같다!!!

[SeongHoonC]

헉 네이버 코테 직전에 풀었던 문제라 다시 풀지는 않았습니다..하하
저는 bfs 로 탐색하고 찾으면 return 하도록 풀었는데 트리로 푸셨군요?
생각지 못한 구현 방법 감사합니다..!

val max = 100_001
fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val visited = MutableList(max) { 0 }
    val queue = ArrayDeque<Int>()
    val (x, y) = br.readLine().split(" ").map { it.toInt() }
    queue.add(x)
    visited[x] = 1

    while (queue.isNotEmpty()) {
        val now = queue.removeFirst()
        if (now == y) {
            println(visited[now]-1)
            return
        }

        val nows = mutableListOf<Int>()

        nows.add(now + 1)
        nows.add(now - 1)
        nows.add(2 * now)

        for (i in nows.indices) {
            if (nows[i] > max - 1 || nows[i] < 0) {
                continue
            }
            if (visited[nows[i]] == 0) {
                visited[nows[i]] = visited[now] + 1
                queue.add(nows[i])
            }
        }
    }
}

[wkdghdwns199]

처음에 어 왜 굳이 result 를 따로 빼서 이동 시간을 저장하지 하고 그냥 쌍을 q에 저장했는데 메모리 초과가 나네요..!

# 메모리 초과 코드
import sys
input=sys.stdin.readline
from collections import deque

N,K=map(int, input().split())

q=deque()
q.append((N,0))
result = 0
while q:
    current,second = q.popleft()
    if current==K:
        result = second
        break
    
    q.append((current-1, second+1))
    q.append((current+1, second+1))
    q.append((current*2, second+1))

print(result)

알고보니 메모리 제한이 128MB 였고 어마무시한 경우의 수가 나올 수 있기 때문에 메모리 초과가 나겠거니 하고 순순히 민석 님처럼 풀었슴다..

import sys
input=sys.stdin.readline
from collections import deque

N,K=map(int, input().split())

q=deque()
q.append(N)
result = [0] * 100001

while q:
    current = q.popleft()
    if current==K:
        break
    
    if current-1 >=0 and result[current-1] ==0:
        result[current-1] = result[current] + 1    
        q.append(current-1)

    if current+1 <=100000 and result[current+1] ==0:
        result[current+1] = result[current] + 1    
        q.append(current+1)

    if current*2 <=100000 and result[current*2] ==0:
        result[current*2] = result[current] + 1    
        q.append(current*2)

print(result[K])

[fnzksxl]

트리로 푸는 방법도 있었군요..
저는 성훈님처럼 BFS로 풀었습니다! 영희가 0~100,000 사이에 있을 수 있어서
Dist 리스트의 크기를 10**6으로 초기화해주고 진행했습니다.

from collections import deque


import sys

input = sys.stdin.readline
INF=10**6

def bfs():
    q = deque()
    q.append(N)
    while q:
        x = q.popleft()
        if x == M:
            print(dist[x])
            break
        for nx in (x-1, x+1, x*2):
            if 0 <= nx < INF and dist[nx] == 0:
                dist[nx] = dist[x] + 1
                q.append(nx)

        
N, M = map(int,input().split())

dist = [0] * (INF)

bfs()