[백준/Python] 15905번 : 스텔라(STELLA)가 치킨을 선물했어요

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

15905번: 스텔라(STELLA)가 치킨을 선물했어요

 

문제

경인지역 6개대학 연합 프로그래밍 경시대회 shake! 는 아주대학교, 경희대학교, 성균관대학교, 인하대학교, 한국항공대학교, 한양대학교ERICA가 함께하는 대학교 자체 연합 대회이다. shake! 는 매년 1등 학생에게 사과북 프로 레티나를 비롯하여 5등까지 상장과 상품을 수여하고 있다.

2018년 올해는 특별히 세계 최고 에듀테크 기업인 ST Unitas와 함께한다. 스텔라(STELLA)는 ST Unitas의 인공지능 사원이다. 인간을 동경하는 스텔라는 최대한 많은 학생에게 혜택이 돌아가지 못하는 점을 늘 안타까워하였다. 혜택을 주기위해 많은 고민을 거듭하던 스텔라는 ST Unitas의 전설적인 “스카이에듀 치킨 기프티콘“ 전략을 읽어보고 감명을 받아 이를 따라하기로 결심하였다. 너무 많은 치킨으로 회사에 큰 피해를 주면 안 되기에, 스텔라는 대회의 5등과 푼 문제 수는 같지만 패널티 차이로 수상하지 못한 학생들에게만 치킨 기프티콘을 보내주고자 한다.

아쉽게도 2018년의 최신기술로도 인공지능 사원이 직접 프로그래밍을 할 수는 없어서, 여러분들이 스텔라를 도와주어야 한다. shake! 순위는 아래와 같은 방식으로 정해진다고 할 때, 스텔라가 몇 개의 기프티콘을 구매해야 하는지 알아보자.

• 각 참가자는 해결한 문제 개수와 패널티 총합을 가진다.
• 해결한 문제의 개수가 더 많은 참가자가 더 높은 순위를 가진다.
• 해결한 문제의 수가 같을 때, 패널티 총합이 더 작은 참가자가 더 높은 순위를 가진다.

입력

첫 줄에 참가자의 수 N(5 ≤ N ≤ 66)이 주어진다. 이후 N개의 줄에 걸쳐 각 참가자가 해결한 문제 개수와 패널티 총합이 주어진다. 각 참가자가 해결한 문제의 개수는 8개보다 작거나 같은 음이 아닌 정수이며 패널티 총합은 100,000보다 작거나 같은 음이 아닌 정수이다. 5등 학생은 적어도 한 문제 이상을 해결하였음이 보장되며, 한 문제 이상을 푼 학생 중 문제 수와 패널티가 모두 같은 학생은 존재하지 않는다.

출력

한 줄에 5등과 해결한 문제 개수가 같지만 수상하지 못하는 학생의 수를 출력한다.

---

코드

n = int(input())
arr = []

for i in range(n):
    a, b = map(int, input().split())
    arr.append([a, b])

arr = sorted(arr, key=lambda x: (-x[0], x[1]))

st = 0
if len(arr) >= 6:
    for i in range(5, len(arr)):
        if arr[4][0] == arr[i][0]:
            st += 1

print(st)

풀이

1. 참가자의 수 n을 입력받는다.

2. 문제 개수와 패널티 점수를 입력받고 2차원 배열에 하나씩 넣는다.

3. 해결한 문제의 개수가 많은 순서대로 먼저 배열하고, 같은 경우 패널티 점수가 낮은 순서대로 배열한다.

4. 5등과 해결한 문제의 개수가 같지만 상을 받지 못하는 학생의 수를 st에 저장한다.

5. st의 값을 출력한다.