[COS Pro 1급, Python] 1차 10번 : 주식으로 최대 수익을 내세요

문제 유형

한줄 수정

난이도

easy

Note 

 Nothing

 

Code

# -*- coding: utf-8 -*-
# UTF-8 encoding when using korean

def solution(prices):
	INF = 1000000001;
	tmp = INF
	answer = -INF
	for price in prices:
		if tmp != INF:
			answer = max(answer, price - tmp)
		tmp = min(tmp, price)
	return answer

prices1 = [1, 2, 3];
ret1 = solution(prices1);

print("solution 함수의 반환 값은", ret1, "입니다.")
    
prices2 = [3, 1];
ret2 = solution(prices2);

print("solution 함수의 반환 값은", ret2, "입니다.")

 

※ 가끔 코드 중 print(~)가 있습니다. 정리 못한 점 죄송합니다.