프그스_스택큐_주식가격 (스택 기본)

RostoryT·2022년 5월 15일
0

Stack Queue

목록 보기
4/17

이 문제에서는 주가가 떨어진 이후 시간들의 주가는 더이상 고려 대상이 아닙니다. 떨어지기 직전까지만 주가가 유지되는 것으로 봅니다.

''' 내가 푼 - 문제 설명 이상하니 아래꺼로 풀어야 함 '''
def solution(prices):
    answer = [0 for _ in range(len(prices))]
    
    for i in range(len(prices)):
        for j in range(i+1, len(prices)):
            if prices[i] > prices[j]:        # 떨어짐
                answer[i] += 1               # 그 직전에 1초는 유지했으므로 +1하고 끝내야함
                break
            else:
                answer[i] += 1
    
    return answer

print(solution([1, 2, 3, 2, 3]))


''' 다른 사람 코드 - 나랑 같음 '''
def solution(prices):
    answer = [0] * len(prices)
    for i in range(len(prices)):
        for j in range(i+1, len(prices)):
            if prices[i] <= prices[j]:
                answer[i] += 1
            else:
                answer[i] += 1
                break
    return answer
profile
Do My Best

0개의 댓글