[baekjoon] 백준 21758번(파이썬): 꿀 따기

문제

21758번: 꿀 따기

알고리즘

- 벌 - 벌 - 꿀인 경우와 꿀 - 벌 - 벌인 경우와 벌 - 꿀 - 벌인 경우를 나누어 문제를 수행한다.

- 벌 - 벌 - 꿀과 꿀 - 벌 - 벌의 경우 첫 번째와 마지막 벌이 먹는 꿀의 양은 고정으로 두고 두 번째 벌이 먹을 수 있는 양을 반복문을 통해 확인하면서 최대 꿀의 양을 비교한다.

- 벌 - 꿀 - 벌의 경우 모든 곳에 꿀을 한 번씩 먹고 꿀통 지점의 꿀을 한번 더 먹었을 때에 값을 반복문을 통해 최대 꿀의 양과 비교한다.

- 3가지 경우를 확인하고 최대의 꿀의 양을 출력한다.

코드

import sys

n = int(sys.stdin.readline())
m = list(map(int, sys.stdin.readline().split()))
target = sum(m[:]) # 전체 꿀의 양
answer = 0
temp = m[0]

# 벌 - 벌 - 꿀인 경우
for i in range(1, n):
    temp += m[i] # 중간 벌이 못 먹는 꿀의 양
    answer = max(answer, target - m[0] + target - temp - m[i]) # 첫 번째 벌이 먹을 수 있는 꿀의 양 + 두번째 벌이 먹을 수 있는 꿀의 양


# 꿀 - 벌 - 벌인 경우
m.reverse()
temp = m[0]
for i in range(1, n):
    temp += m[i] # 중간 벌이 못 먹는 꿀의 양
    answer = max(answer, target - m[0] + target - temp - m[i]) # 첫 번째 벌이 먹을 수 있는 꿀의 양 + 두번째 벌이 먹을 수 있는 꿀의 양


# 벌 - 꿀 - 벌인 경우
for i in range(1, n):
    answer = max(answer, target - m[0] - m[-1] + m[i]) # 모든 곳에 꿀을 한번씩 먹고 꿀통 지점의 꿀을 한번 더 먹는다.


print(answer)

github

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법