20-Hwangyerin

[Hwangyerin]

🔗 문제 링크

Baekjoon | 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보다 작거나 같다.

✔️ 소요된 시간

30분

✨ 수도 코드

Note

유제품 3개를 한 번에 산다면 그중에서 가장 싼 것은 무료 로 지불하고 나머지 두 개의 제품 가격만 지불하면 됩니다.
한 번에 3개의 유제품을 사지 않는다면 할인 없이 정가를 지불해야 합니다.

N개의 유제품을 모두 살 때 필요한 최소비용 을 구하기 위해서는 최대한 가격이 비싼 유제품을 무료 로 지불해야한다.
즉, 입력 받은 값을 내림차순으로 정렬해 주었다.

처음 코드

1. 입력 받은 유제품의 가격들을 리스트로 입력받아 내림차순으로 정렬한다.
2. 길이가 3인 부분 리스트를 생성하고 그 안의 최솟값 (min()) 을 길이가 3인 부분 리스트의 합(sum()) 에서 뺀 후 총 합에 더해준다.

ans += sum(tmp) - min(tmp)

3. 만약 부분 리스트의 길이가 3보다 작을 수 있기 때문에 그런 경우에는 2+1 행사를 진행할 수 없으므로 정가를 계산한다. (sum())

ans+=sum(tmp)

전체코드

import sys
input = sys.stdin.readline
N = int(input())
dairy_product = [int(input()) for _ in range(N)]
dairy_product = sorted(dairy_product,reverse=True)
ans = 0
for i in range(0,N,3):
    tmp = dairy_product[i:i+3]
    if len(tmp) < 3:
        ans+=sum(tmp)
    else:
        ans += sum(tmp) - min(tmp)
print(ans)

문제가 맞긴 했으나 코드가 마음에 들지 않고 더 최적화 시킬 수 있을 것 같아 수정해 보았다.

두번째 코드

1. 입력 받은 유제품 가격들을 내림차순으로 정렬하여 리스트 생성한다.
2. 전체 리스트의 합을 초기값으로 설정
3. 리스트를 세 번째 요소부터 시작하여 세 개씩 건너뛰면서 현재 요소의 값을 전체 합에서 뺀다.

우선, 첫번째 코드와 다른 점은, 유제품의 가격을 입력을 받고 읽으면서 동시에 리스트를 생성하고 정렬한다.

두번째로, 합계를 계산하는 과정에서 리스트를 전체를 훑지 않는다. 미리 전체 요소의 합을 계산하고, 일정한 규칙에 따라 일부 값을 빼서 최소비용을 계산한다.

import sys
input = sys.stdin.readline
N = int(input())

# 입력 받은 유제품 가격들을 내림차순으로 정렬하여 리스트 생성
dairy_product = sorted([int(input()) for _ in range(N)],reverse=True)

# 유제품 가격 리스트의 합을 초기값으로 설정
ans = sum(dairy_product)

# 유제품 가격 리스트의 합에서 세 번째 요소부터 시작하여 세 개씩 건너뛰면서 현재 요소의 값을 빼는 과정
for i in range(2,N,3):
    ans-= dairy_product[i]

# 필요한 최소비용 출력
print(ans)

코드를 수정하니 시간이 112ms 에서 92ms로 줄었다.

풀었던 코드도 다시 돌아보자!

📚 새롭게 알게된 내용

[janghw0126]

저도 예린님이 처음으로 짜신 코드와 비슷하게 짰는데 되돌아보지 않고 넘겼던 이력이 있습니다,,,😅 반성해야겠네요😥😥
덕분에 새로운 풀이 방법도 알아갑니다!
저두 풀었던 코드도 다시 한번 돌아보는 습관을 들여야겠네요🔥🔥

이번 PR도 수고하셨어요!🍬

[avocado-13]

저는 초기값을 전체 유제품 가격의 합으로 두고, 빼는 당식으로 진행하였는데 정답이었으나 예린님의 1차시도 처럼 112 m가 뜨더군요. 2차 코드처럼 저렇게 빠른 속도로 실행되는 코드도 있다는 것을 알아갑니다. 또, 코드 최적화의 중요성도 알아가요 !! 💞이번 차시 PR도 수고많으셨습니다 👍🏻👍🏻

[LJEDD2]

리스트 슬라이싱을 사용하여 for 루프 대신에 연산을 수행해볼 수도 있겠네요 🎃
늘 꼬롬한 코드를 개선시켜보려는 노력 정말 보기 좋습니다 !!

ans -= sum(dairy_product[2:N:3])