규식이는 친구들 사이에서 투자의 귀재로 알려져 있습니다. 페이수북과 인수타그램에 자신의 성과를 과시하기 때문인데요. 사실 규식이가 그 정도의 실력자는 아닙니다. 성과가 좋을 때에만 SNS에 공유해서 그렇게 비춰질 뿐이죠.
계속해서 멋진 모습을 보여주기 위해, 특정 기간 중 수익이 가장 큰 구간을 찾아내는 함수 sublist_max를 작성해 보려고 합니다.
Brute Force 방법을 이용해서 이 문제를 한 번 풀어봅시다!
우선 함수 sublist_max는 파라미터로 리스트 profits를 받는데요. profits에는 며칠 동안의 수익이 담겨 있습니다. 예를 들어서 profits가 [7, -3, 4, -8]이라면 첫 날에는 7달러를 벌었고, 둘째 날에는 3달러를 잃었고, 셋째 날에는 4달러를 벌었고, 마지막 날에는 8달러를 잃은 거죠.
sublist_max 함수는 profits에서 최대 수익을 내는 구간의 수익을 리턴합니다.
profits가 [7, -3, 4, -8]이라면 무엇을 리턴해야 할까요? profits에서 가장 많은 수익을 낸 구간은 [7, -3, 4]입니다. 이 구간에서 낸 수익은 8달러이니, 8을 리턴하면 되겠죠!
만약 profits가 [-2, -3, 4, -1, -2, 1, 5, -3]이라면? profits에서 수익이 가장 큰 구간은 [4, -1, -2, 1, 5]입니다. 이 구간에서 낸 수익은 7달러이니, 7을 리턴하겠죠?
내 코드:
def sublist_max(profits):
# 코드를 작성하세요.
result = 0
new_sublist = [profits[0]]
for i in range(1, len(profits)):
if profits[i] * new_sublist[len(new_sublist)-1] > 0:
new_sublist[len(new_sublist)-1] += profits[i]
else:
new_sublist.append(profits[i])
if new_sublist[0] < 0:
new_sublist.reverse()
if (len(new_sublist)+1)//2 == 1:
result = new_sublist[0]
for j in range(0, len(new_sublist), 2):
for k in range(0, j+1, 2):
temp = sum(new_sublist[k:j+1])
if temp > result:
result = temp
return result
# 테스트
print(sublist_max([4, 3, 8, -2, -5, -3, -5, -3]))
print(sublist_max([2, 3, 1, -1, -2, 5, -1, -1]))
print(sublist_max([7, -3, 14, -8, -5, 6, 8, -5, -4, 10, -1, 8]))