[Sort] Boj10825: 국영수

[Sort] Boj10825: 국영수

Link: https://www.acmicpc.net/problem/10825

문제

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

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

입력

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

출력

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

입출력 예제

코드

import sys
si = sys.stdin.readline

score = [[] for _ in range(N)]
#입력 받아주기(이름, 국어, 영어, 수학)
N = int(si())
for i in range(N):
	score[i] = list(si().split())
	score[i][1:] = map(int,score[i][1:])

#정렬(Key 값의 순서 중요)
score.sort(key = lambda x:(-x[1],x[2],-x[3],x[0]))

for x in score:
	print(x[0])

코드 간략 설명

1. 이름과 성적을 입력받아주고 성적은 int로 변환하여 다시 저장해준다.
2. 정렬시 순서가 중요한데, 키값을 4가지로 설정하여 정렬한다.
   2-1. 가장 우선순위가 국어 성적이다. 국어 성적이 큰 순서부터 나열하면 되기 때문에 key 값 'x[1]' 앞에 마이너스 부호를 붙여 감소 순서대로 정렬해준다.
   2-2. 국어성적과 마찬가지로, 다음은 영어, 수학, 이름 순으로 차례대로 key 값을 설정해 주면 된다.
3. 정렬된 리스트의 이름만 출력 해준다.

Screenshot & Output