백준 11508번 Python / Greedy

문제풀이 GitHub

https://github.com/Seokii/baekjoon

GitHub - Seokii/baekjoon: Daily Commit for Baekjoon

 

문제 링크

https://www.acmicpc.net/problem/11508

11508번: 2+1 세일

 

풀이

n = int(input())
arr = []
answer = 0
cnt = 0
for _ in range(n):
    arr.append(int(input()))

arr.sort(reverse=True)

for i in range(len(arr)):
    if cnt != 2:
        cnt += 1
        answer += arr[i]
    elif cnt == 2:
        cnt = 0

print(answer)

첫 줄에서 n의 값을 입력하고, n번 만큼 가격의 정보를 받아 arr변수에 리스트를 선언하고 저장했습니다.

리스트를 내림차순으로 정렬하고 cnt변수를 주어 3의 배수만큼의 인덱스는 제외하고 총합을 구해 출력했습니다.