[python 파이썬][프로그래머스] 탑(스택/큐)

https://programmers.co.kr/learn/courses/30/lessons/42588

코딩테스트 연습 - 탑

- 문제 설명

수평 직선에 탑 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]

---

◆ 문제풀이

• 먼저 레이저가 오른쪽으로 쏴질 수 있도록 탑의 순서를 거꾸로 바꾸어 줍니다.(왼쪽->오른쪽으로 생각하는 것이 좀더 편리하기 때문입니다.)
• 하나씩 stack에 적재하고 이후의 탑의 높이와 비교해서 적재한 탑보다 높은 숫자의 탑이 나오면 해당 탑의 위치를 기록하고, 높은 숫자의 탑이 나오지 않으면 0을 기록한다.

 

◆ 코드 풀이

def solution(heights):
    answer = []
    stack = []
    heights_rev = heights[::-1]
    
    for i in range(len(heights_rev)):
        stack.append(heights_rev[i]) #stack에 하나씩 적재
        for j in range(i+1,len(heights_rev)): #적재한것 뒤부터 비교
            if stack[-1] < heights_rev[j]: #어떤 탑이 적재한 것보다 숫자가 작으면
                answer.append(len(heights_rev)-j) #해당 탑의 위치를 기록하고 중지
                break
        if len(answer) != i+1: #위의 과정에서 아무것도 추가되지 않았다면
            answer.append(0) #신호를 쏜 탑보다 숫자가 큰 것이 없었으므로 0을 추가
                
    return answer[::-1] #원래대로 순서를 거꾸로 바꿈