프로그래머스_스택_탑

적당한 시간내에 적당히 잘 푼것같은 문제이다. 근데 효율성 측면은 잘 모르겠다.

---

문제

수평 직선에 탑 N대를 세웠습니다. 모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다. 발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다. 또한, 한 번 수신된 신호는 다른 탑으로 송신되지 않습니다.

예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다. 그러면, 탑은 다음과 같이 신호를 주고받습니다. 높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고, 높이가 7인 네 번째 탑의 신호는 높이가 9인 두 번째 탑이, 높이가 5인 세 번째 탑의 신호도 높이가 9인 두 번째 탑이 수신합니다. 높이가 9인 두 번째 탑과 높이가 6인 첫 번째 탑이 보낸 레이저 신호는 어떤 탑에서도 수신할 수 없습니다.

 

나의 풀이

def solution(heights):
    answer = []
    for i, h in enumerate(heights):
        if max(heights[0:i+1]) == h:
            answer.append(0)
        else:
            k = len(heights[0:i+1])
            while True:
                if heights[0:i+1][k-1] > h:
                    answer.append(k)
                    break;
                k -= 1

    return answer

for문안에 while문을 만들고, 했지만 가독성이 나쁜것 같지는 않다.

전체적인 로직은, 일단 리스트의 왼쪽부터 탐색하는데 만약 탐색하는 수 (for문의 h)가

슬라이싱의 최대값이었다면(리스트 슬라이싱을 이용) 0을 append해준다.

그리고 만약 아니라면?

그 값부터 while문을 활용해서 다시 오른쪽으로 탐색한다. 근데 만약 탐색한 값이 h보다 크면? 바로 while문을 break하고 answer에 인덱스를 append 해주면 된다!

 

다른 사람의 풀이

def solution(h):
    ans = [0] * len(h)
    for i in range(len(h)-1, 0, -1):
        for j in range(i-1, -1, -1):
            if h[i] < h[j]:
                ans[i] = j+1
                break
    return ans

for문 두개를 자유자재로 쓰는것도 신기하지만, 

ans = [0] * len(h) 의 잡기술을 한수 배워간다..