[백준 11508 파이썬] 2+1 세일

문제

KSG 편의점에서는 과일우유, 드링킹요구르트 등의 유제품을 '2+1 세일'하는 행사를 하고 있습니다. KSG 편의점에서 유제품 3개를 한 번에 산다면 그중에서 가장 싼 것은 무료로 지불하고 나머지 두 개의 제품 가격만 지불하면 됩니다. 한 번에 3개의 유제품을 사지 않는다면 할인 없이 정가를 지불해야 합니다.

예를 들어, 7개의 유제품이 있어서 각 제품의 가격이 10, 9, 4, 2, 6, 4, 3이고 재현이가 (10, 3, 2), (4, 6, 4), (9)로 총 3번에 걸쳐서 물건을 산다면 첫 번째 꾸러미에서는 13원을, 두 번째 꾸러미에서는 10원을, 세 번째 꾸러미에서는 9원을 지불해야 합니다.

재현이는 KSG 편의점에서 친구들과 같이 먹을 총 N팩의 유제품을 구입하려고 합니다. 재현이를 도와 최소비용으로 유제품을 구입할 수 있도록 도와주세요!

입력

첫 번째 줄에는 유제품의 수 N (1 ≤ N ≤ 100,000)이 주어집니다.

두 번째 줄부터 N개의 줄에는 각 유제품의 가격 Ci (1 ≤ Ci ≤ 100,000)가 주어집니다.

출력

재현이가 N개의 유제품을 모두 살 때 필요한 최소비용을 출력합니다. 정답은 231-1보다 작거나 같다.

풀이

각 제품을 3개의 세트로 샀을 때 그 중 가장 싼 제품은 무료로 살 수 있다.

따라서 입력값인 제품의 가격을 내림차순으로 정렬한다.

정렬한 입력값을 반복할 때 3번째 마다 돈을 지불하지 않아도 된다.

그리고 처음 3번째 인덱스는 2이다.

따라서 k의 초기값을 2로 설정하고, i가 k일 때는 ans에 값을 더하지 않고 다음 행사 제품을 위해 k에 3을 더해준다.

반복문이 끝나고 최종 ans값을 출력한다.

 

import sys
n = int(sys.stdin.readline().rstrip("\n"))
nums=[int(sys.stdin.readline().rstrip("\n")) for _ in range(n)]
nums = sorted(nums, reverse=True)
k=2
ans=0
for i in range(n):
    if(i==k):
        k+=3
        continue
    ans+=nums[i]
print(ans)