[python] 백준 21758번: 꿀 따기(2021 정보올림피아드 중등부 2교시 1번)

problem

문제링크: https://www.acmicpc.net/problem/21758

입력 : N, 각 장소의 꿀의 양

출력 : 최대의 꿀의 양

solution

두 꿀벌의 위치와 벌통의 위치를 3가지로 나눠 각 경우중 가장 큰 값 출력

1. 벌통이 오른쪽 끝에 있는경우
2. 벌통이 왼쪽 끝에 있는경우
3. 벌통이 두 벌 사이에 있는 경우

code

n = int(input())
ans=0
a = list(map(int,input().split()))
s=[]
s.append(a[0])

for i in range(1,n):
    s.append(s[i-1]+a[i])

for i in range(1,n-1):
    ans=max(ans,s[n-2]-a[0]+a[i])

for i in range(1, n - 2):
    ans = max(ans, s[n-1] - a[0] - a[i] + s[n-1] - s[i])

for i in range(1,n-1):
    ans=max(ans,2*s[i-1]+s[n-2]-s[i])
    
print(ans)