본문 바로가기

알고리즘/프로그래머스(Python)

[알고리즘] 프로그래머스 탑 / python

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

def solution(heights):
    answer = [0] * len(heights)
    heights = heights[::-1]
    
    for i in range(len(heights)):
        for j in range(i+1, len(heights)):
            if heights[j] > heights[i]:
                answer[i] = len(heights)-j
                break
                
    return answer[::-1]

 

더 간결하게 푼 코드가 있었다 ...

 

def solution(heights):
    answer = [0] * len(heights)
    
    for i in range(len(heights)-1, 0, -1):
        for j in range(i-1, -1, -1):
            if heights[j] > heights[i]:
                answer[i] = j + 1
                break
                
    return answer