스택/큐 (Level 2) : 주식가격 (Python)

문제 : 

코딩테스트 연습 - 주식가격

 

스택/큐에 있던 카테고리지만 그냥 이중 for문으로 풀었다.

 

입출력 예시의 3초, 4초 시점에서 가격이 1초 뒤에 떨어져도, 1초간 가격이 떨어지지 않아도

둘다 1초간 가격이 떨어지지 않은 경우라는 것이 직관적으론 이해가 갔는데 

처음 구현할때는 계속 원하는 출력이 나오지 않았다 ㅡㅡ

 

짜증나서 이틀 뒤에 처음부터 다시 풀었더니 풀렸다.

 

Solution

def solution(prices):
    answer = []
    size=len(prices)
    for i in range(size):
        cnt=1
        for j in range(i+1,size-1):        
            if prices[i]>prices[j]: break    
            cnt+=1
            
        answer.append(cnt)
        if len(answer)==size-1:
            answer.append(0)
            return answer

 

헤헤