취준/프로그래머스

[프로그래머스 Programmers][Python] 탑

puff 2020. 1. 17. 22:00

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

 

코딩테스트 연습 - 탑 | 프로그래머스

수평 직선에 탑 N대를 세웠습니다. 모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다. 발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다. 또한, 한 번 수신된 신호는 다른 탑으로 송신되지 않습니다. 예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다. 그러면, 탑은 다음과 같이 신호를 주고받습니다. 높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고, 높이가 7

programmers.co.kr

이 문제도 간단하다. 어차피 신호는 오른쪽 에서 왼쪽으로 쏘니까 탑 하나하나 돌면서 왼쪽거 확인하면 된다.

 

주의할점

  • 신호가 왼쪽으로 가다가 수신하면 그걸로 끝, break 해야함
  • 탑 번호는 1부터 시작함. ret[t] = n+1 해야하는 이유
def solution(heights):

    ret = [0]*len(heights)
    for t in range(len(heights)):
        tlen = heights[t]

        for n in reversed(range(t)):
            if heights[n] > tlen:
                ret[t] = n+1
                break


    return ret