[Algorithm] [Python] 백준/BOJ - 10825 _ 국영수

10825 - 국영수

문제

---

도현이네 반 학생 N명의 이름과 국어, 영어, 수학 점수가 주어진다. 이때, 다음과 같은 조건으로 학생의 성적을 정렬하는 프로그램을 작성하시오.

• 국어 점수가 감소하는 순서로
• 국어 점수가 같으면 영어 점수가 증가하는 순서로
• 국어 점수와 영어 점수가 같으면 수학 점수가 감소하는 순서로
• 모든 점수가 같으면 이름이 사전 순으로 증가하는 순서로 (단, 아스키 코드에서 대문자는 소문자보다 작으므로 사전순으로 앞에 온다.)

입력

---

첫째 줄에 도현이네 반의 학생의 수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 한 줄에 하나씩 각 학생의 이름, 국어, 영어, 수학 점수가 공백으로 구분해 주어진다. 점수는 1보다 크거나 같고, 100보다 작거나 같은 자연수이다. 이름은 알파벳 대소문자로 이루어진 문자열이고, 길이는 10자리를 넘지 않는다.

출력

---

문제에 나와있는 정렬 기준으로 정렬한 후 첫째 줄부터 N개의 줄에 걸쳐 각 학생의 이름을 출력한다.

제출 코드

---

from sys import stdin
n = int(stdin.readline())
arr = [ list(stdin.readline().split()) for _ in range(n)]
arr = sorted(arr, key = lambda x: (-int(x[1]), int(x[2]), -int(x[3]), x[0]))
for item in arr: 
    print(item[0])

풀이 설명

---

위 문제는 다양한 조건을 적용한 정렬을 할 줄 아는가를 묻는 문제이다.

총 4개의 조건에 따라 입력값을 정렬한 뒤 이를 출력하면 된다.

파이썬에서는 lambda 식을 이용해 간편하게 다중 조건 정렬을 실행할 수 있다.

조건이 여러개인 경우, lambda의 인자값을 튜플로 묶고 내림차순의 경우 - 부호를 붙여주면 된다.

따라서 위 문제의 조건 4개를 만족하는 정렬식을 아래 한 줄로 구현이 가능하다.

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