[프로그래머스] 주식 가격 풀이

주식가격 문제

스택

Last In First Out(LIFO), 나중에 들어간 원소가 먼저 나옴 push: 스택에 쌓음 pop: 가장 마지막에 들어간 원소를 제거하며 반환

파이썬 풀이

def solution(prices): st=[] answer = [0]*len(prices) for i in range(len(prices)): if st == []: st.append(i) continue else: if prices[i] >= prices[i-1]: st.append(i) else: while prices[i] < prices[i-1] : top = st.pop() answer[top] = i-top if st == []: st.append(i) break top = st.pop() st.append(top) if prices[i] >= prices[top] : st.append(i) break top = st.pop() st.append(top) while st != []: n=st.pop() answer[n] = top - n return answer

그런데 별로.. 좋은 코드라고 생각은 안 들지만 어쨌든 해결

참고한 테스트 케이스

prices = [3, 1, 2, 4, 2]
answer = [1, 3, 2, 1, 0]

풀이 방법 (https://gurumee92.tistory.com/m/170 참고)