[백준] 10825: 국영수 (Python)

국영수

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

10825번: 국영수

 

문제

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

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

 

입력

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

 

출력

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

 

입출력 예제

입력	출력
12 Junkyu 50 60 100 Sangkeun 80 60 50 Sunyoung 80 70 100 Soong 50 60 90 Haebin 50 60 100 Kangsoo 60 80 100 Donghyuk 80 60 100 Sei 70 70 70 Wonseob 70 70 90 Sanghyun 70 70 80 nsj 80 80 80 Taewhan 50 60 90	Donghyuk Sangkeun Sunyoung nsj Wonseob Sanghyun Sei Kangsoo Haebin Junkyu Soong Taewhan

 

풀이

우선순위가 다른 4개의 기준에 대해 정렬하면 된다. 람다식을 이용한 다중정렬을 통해 해결할 수 있다.

정렬 순서: 국어 내림차순 → 영어 오름차순 → 수학 내림차순 → 이름 오름차순

import sys
input = sys.stdin.readline
n = int(input())
info = []
for _ in range(n):
	info.append(input().split())
info.sort(key = lambda x: (-int(x[1]), int(x[2]), -int(x[3]), x[0]))
for student in info:
	print(student[0])

메모리: 86852 KB / 시간: 416 ms

 

info 리스트를 구성할 때 input().split()으로 넣는 것보다 list(input().split())으로 넣었을 때 메모리 사용량이 더 적게 나타났다. split() 함수가 리스트를 반환하므로 둘의 실행 결과 자체는 리스트로 동일한데 … 왜 메모리 사용량에 차이가 나는지 정확한 이유를 아직 모르겠다.

또한 print 대신 sys.stdout.write를 쓰면 실행 시간을 단축시킬 수 있다.

import sys 
input = sys.stdin.readline
n = int(input())
info = [list(input().split()) for _ in range(n)]
info.sort(key = lambda x: (-int(x[1]), int(x[2]), -int(x[3]), x[0]))
for student in info:
	sys.stdout.write(str(student[0]) + "\\n")

메모리: 80708 KB / 시간: 384 ms