[프로그래머스 스택/큐]탑 - Python3

문제

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]

 

알고리즘

def solution(heights):
	# 정답 배열 0으로 초기화
    answer = [0] * len(heights)
    
    # i가 배열의 맨 뒤에서부터 시작해서 1씩 감소할 것이다. 
    i = len(heights) - 1
    while i > 0:
        temp = heights[i]
        j = i - 1
        
        # j는 i -1 부터 1씩 감소해서 0까지 반복할 것이다. 
        while j >= 0:
        	# height[i]가 height[j]보다 작아면 그때 신호가 수신된다.
            if temp < heights[j]:
                answer[i] = j+1
                break
            # height[i]가 height[j]보다 크다면 신호가 수신되지 않는다.
            else:
                j -= 1
        i -= 1
    return answer


heights =[6,9,5,7,4]
print(solution(heights))

설명

그냥 2차원 배열 돌리면서 i랑 j랑 크기 비교하다가 j번째 값이 더 크면 answer[j+1]번째에 그 값 넣어주면 된다. 

결과

후기

스택이나 큐 등의 자료구조를 이용해서 풀려고 했는데, 딱히 리스트 말고 다른 자료구조를 써서 풀어야 할 필요를 못느꼈다. 다른 사람들도 다 2차원 배열로 풀었더라. 이러면 heights배열의 길이가 만 십만 이렇게 되면 답 안나올텐데 어떻게 풀까. 항상 그것을 생각하면서 풀어야 하는데 뭐 어쩔지 아직 잘모르게따~~~~~~~~~~