30-xxubin04

[xxubin04]

🔗 문제 링크

백준 14562번: 태권왕🏋🏻‍♂️

✔️ 소요된 시간

1시간 30분?

✨ 수도 코드

1. 문제 이해

태균이의 연속 발차기 2가지

1. 태균이의 점수 * 2이면서 상대의 점수 + 3
2. 태균이의 점수 + 1

태균이의 점수와 상대의 점수가 주어질 때, 두 사람의 점수가 동일해지도록 하는 연속 발차기의 최소 횟수를 구하는 것이다.

2. DFS? BFS?

처음에 문제를 보고 DFS와 BFS 중 무엇으로 풀어야 할지 감을 잡지 못했다. 그새 까먹었나보다... 그래서 DFS로 도전하다가 아무리 코드를 수정해보아도 원하는 값이 나오질 않아 BFS로 다시 코드를 짰다. BFS 알고리즘이 기억이 잘 안나 블로그에서 아이디어를 참고하였다.

BFS는 너비 우선 탐색이다. 즉, 두 노드 사이의 최단 경로를 구하거나 임의의 경로를 찾고 싶을 때 이 방법을 택한다. 이 문제는 임의의 경로를 찾고 싶을 때와 유사하다. 태균이의 처음의 점수에서 어떠한 연속 발차기를 택하게 되는 경로에 따라 값이 달라지기 때문이다.

반대로, DFS는 깊이 우선 탐색이다. 모든 노드를 방문하고자 하는 경우에 이 방법을 택한다. DFS는 다음 브랜치를 탐색하기 전에 해당 브랜치를 완벽하게 탐색하고 넘어가는 방법이며, 자기 자신을 호출하는 순환 알고리즘이다.

3. 코드 분석

from collections import deque
input = open(0).readline 

def BFS(c, x, y):
    global ans
    q = deque()
    q.append((c, x, y))
    while q:
        cnt, t, o = q.popleft()
        if t == o:
            ans.append(cnt)
            return ans
        elif t * 2 <= o + 3:
            q.append((cnt+1, t*2, o+3))
            q.append((cnt+1, t+1, o))
        else:
            q.append((cnt+1, t+1, o))

BFS의 특징은 deque를 사용하는 것이다. 그러므로 collections 모듈에서 deque를 import해준다.

c : 이전 연속 발차기 과정에서의 cnt 매개변수 | cnt
x : 태균이의 현재 점수 매개변수 | t
y : 상대의 현재 점수 매개변수 | o
ans : '태균 점수 == 상대 점수'가 되는 여러 경로에서의 cnt가 모인 리스트

• 태균 점수 == 상대 점수
  • ans에 cnt를 추가해주고 ans를 반환한다.
• 태균 점수 * 2 <= 상대 점수 + 3
  • 우리는 태균 점수와 상대 점수가 같아지도록 해야한다. 하지만, 태균 점수가 상대 점수보다 높아지게 되는 경우가 단 한번이라도 있으면 절대 같아질 수 없다. 돌아올 수 없는 강을 건넌 것이다.
  • 위에서 언급한 첫 번째 연속 발차기를 하게 되었을 때, 태균 점수가 상대 점수보다 작거나 같은 경우에만 첫 번째 연속 발차기를 해준다. 그리고 첫 번째 연속 발차기를 할 수 있는 경우는 태균 점수 + 1가 상대 점수보다 무조건 작거나 같은 경우이다. (태균 점수 + 1 <= 상대 점수) 그러므로, 위의 태균 점수 * 2 <= 상대 점수 + 3에 두 가지 연속 발차기를 해주는 경로를 다 고려해야 한다.
• 그 외의 경우는 두 번째 연속 발차기만 해줄 수 있다.
  • 문제에서 (1 <= 태균 점수 <= 상대 점수 <= 100)으로 주어졌으므로 태균 점수가 상대 점수보다 커지는 경우는 생각하지 않아도 된다.

for i in range(n := int(input())):
    ans = []
    tae, opp = map(int, input().split())
    print(min(BFS(0, tae, opp)))

n번만큼 연속 발차기의 최소 횟수를 구해주어야 하므로 for문 안에 ans = []를 써주어 새로운 태권도 겨루기를 할 때마다 ans가 갱신되도록 해야한다.

위에서 구현한 BFS함수는 ans를 반환하므로 min(BFS(0, tae, opp)) == min(ans)이다. 그러므로, 여러 경우의 cnt들을 담고 있는 ans 리스트에서 최솟값을 출력한다.

4. 전체 코드

from collections import deque
input = open(0).readline 

def BFS(c, x, y):
    global ans
    q = deque()
    q.append((c, x, y))
    while q:
        cnt, t, o = q.popleft()
        if t == o:
            ans.append(cnt)
            return ans
        elif t * 2 <= o + 3:
            q.append((cnt+1, t*2, o+3))
            q.append((cnt+1, t+1, o))
        else:
            q.append((cnt+1, t+1, o))
    
for i in range(n := int(input())):
    ans = []
    tae, opp = map(int, input().split())
    print(min(BFS(0, tae, opp)))

📚 새롭게 알게된 내용

BFS와 DFS를 다시 익혔습니다.
까먹지 않도록 계속 풀어야겠습니다..😂

[Dolchae]

저번에 고민이라고 하셨던 게 이런 말씀이셨군요! 저도 아직 BFS나 DFS에 대해 잘 알지 못해서 도움을 드리긴 어렵지만.. 함께 달려봅시다!!😄 수고하셨어요😊

[9kyo-hwang]

우선순위 큐(파이썬에서는 heapq)를 사용해서 푼 코드가 있더라구요?
내 점수(s)와 상대 점수(t)의 차이가 가장 작은 걸 top에 올리는 원리로 heapq를 쓰네요...

from heapq import heappush, heappop
input = open(0).readline

for _ in range(C := int(input())):
    S, T = map(int, input().split())
    heap = [(0, S, T)]  # 점수 차이, 내 점수, 상대 점수
    
    while heap:
        cnt, s, t = heappop(heap)  # 점수 차이가 작은 순으로 top에 올라옴
        
        if s == t:
            print(cnt)
            break
        
        for ns, nt in [(s * 2,  t + 3), (s + 1, t)]:
            if ns > nt:
                continue
            
            heappush(heap, (cnt + 1, ns, nt))

[9kyo-hwang]

while문에서 처음으로 t == o에 걸리는 경우는 자연스럽게 최단 경로 입니다. 그것이 bfs이기 때문이죠.
그렇기 때문에 리스트에 넣을 필요 없이 즉시 cnt를 반환하는 것으로도 충분합니다.

if t == 0:
    return cnt

[9kyo-hwang]

요 부분은 이렇게 한다기 보단, 어차피 t * 2 케이스와 t + 1 케이스 모두 고려해봐야 하잖아요?
그렇다면 elif - else 보단 각각 if로 걸어버리는 것이 낫습니다. 위의 조건문을 수행한 것과 별개로 아래 조건문도 확인해봐야 하니까요 :)

if t * 2 <= o + 3:
    q.append((cnt + 1, t * 2, o + 3))

if t + 1 <= o:
    q.append((cnt + 1, t + 1, o))

[9kyo-hwang]

첫 번째 코멘트에서 cnt를 즉시 반환받았으니, 여기서는 그 값을 출력만 하면 되겠죠?

print(BFS(0, tae, opp))