출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges
Challenge
문제 설명
수평 직선에 탑 N대를 세웠습니다. 모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다. 발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다. 또한, 한 번 수신된 신호는 다른 탑으로 송신되지 않습니다.
예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다. 그러면, 탑은 다음과 같이 신호를 주고받습니다. 높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고, 높이가 7인 네 번째 탑의 신호는 높이가 9인 두 번째 탑이, 높이가 5인 세 번째 탑의 신호도 높이가 9인 두 번째 탑이 수신합니다. 높이가 9인 두 번째 탑과 높이가 6인 첫 번째 탑이 보낸 레이저 신호는 어떤 탑에서도 수신할 수 없습니다.
| 송신 탑(높이) | 수신 탑(높이) |
|---|---|
| 5(4) | 4(7) |
| 4(7) | 2(9) |
| 3(5) | 2(9) |
| 2(9) | - |
| 1(6) | - |
맨 왼쪽부터 순서대로 탑의 높이를 담은 배열 heights가 매개변수로 주어질 때 각 탑이 쏜 신호를 어느 탑에서 받았는지 기록한 배열을 return 하도록 solution 함수를 작성해주세요.
제한사항
- heights는 길이 2 이상 100 이하인 정수 배열입니다.
- 모든 탑의 높이는 1 이상 100 이하입니다.
- 신호를 수신하는 탑이 없으면 0으로 표시합니다.
입출력 예
| heights | return |
|---|---|
| [6, 9, 5, 7, 4] | [0, 0, 2, 2, 4] |
| [3, 9, 9, 3, 5, 7 ,2] | [0, 0, 0, 3, 3, 3, 6] |
| [1, 5, 3, 6, 7 ,6, 5] | [0, 0, 2, 0, 0, 5, 6] |
입출력 예 설명
입출력 예 #2
[1,2,3] 번째 탑이 쏜 신호는 아무도 수신하지 않습니다.
[4,5,6] 번째 탑이 쏜 신호는 3번째 탑이 수신합니다.
[7] 번째 탑이 쏜 신호는 6번째 탑이 수신합니다.
입출력 예 #3
[1,2,4,5] 번째 탑이 쏜 신호는 아무도 수신하지 않습니다.
[3] 번째 탑이 쏜 신호는 2번째 탑이 수신합니다.
[6] 번째 탑이 쏜 신호는 5번째 탑이 수신합니다.
[7] 번째 탑이 쏜 신호는 6번째 탑이 수신합니다.
Solution
1. On my own
def solution(heights):
"""
- param heights : 맨 왼쪽부터 순서대로 탑의 높이 배열 (List)
- return : 각 탑이 쏜 신호를 어느 탑에서 받았는지 기록한 배열 (List)
"""
from collections import deque
answer = deque()
heights = deque(h for h in heights)
while heights :
top = heights.pop()
left_len = len(heights)
while left_len :
if heights[left_len -1] > top :
break
else :
left_len -= 1
answer.appendleft(left_len)
return list(answer)풀이 방법
collections.deque를 생성후, pop을 통해 가장 오른쪽 탑과 나머지 탑을 역순으로 비교함.
2. Best Code
def solution(heights):
answer = [0]*len(heights)
stack = []
for i in reversed(range(len(heights))):
while stack and stack[-1][1] < heights[i]:
idx, height = stack.pop()
answer[idx] = i+1
stack.append((i, heights[i]))
print(stack)
return answer풀이 방법
대부분의 풀이를 보면 나처럼 heights의 가장 오른쪽 탑을 뽑아서 왼쪽으로 이동하며 비교하는데, 이 코드는 역발상으로 진행한다.
처음엔 잘 이해가 되지 않았는데, 직접 반복문을 하나씩 실행해보니 while문 안에서 오른쪽으로 이동한다.
Lessons learned
역으로 푸는 Best Code에 감탄하지만, 또 하나 알게된 점이 있다.
while (stack and stack[-1][1] < heights[i]) 의 코드에서 stack = []인 상태에선 stack[-1][1]에서 오류가 날 줄 알았는데.. 안난다는 점이다. 알아보니, and 앞에 이미 False로 논리값이 들어갔기 때문에 and 뒤에는 실행이 안된다는 것이다!!!
(나만 몰랐나..?? ㅎㅎ.. 갈길이 멀구만! )
