[Python] 주식가격

 

[코딩테스트 고득점 KIT - 스택/큐]

 

 

이중 for문을 이용하여

기준으로 둔 i번째 가격이 그 다음에 나올 j초 후 가격과 비교하며

cnt 값을 증가시켰다.

 

마지막 값과 비교를 마치거나, 기준 가격보다 떨어진 경우

answer 값을 반환해준다.

 

마지막 가격은 비교할 대상이 없으므로 제외 시켰다가,

마지막에 0 값을 대입해주었다.

 

끝

 

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