[백준] 10825번 국영수

www.acmicpc.net/problem/10825

10825번: 국영수

 

문제

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

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

 

입력

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

 

출력

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

 

 

풀이

파이썬의 lambda 기능을 이용하여 여러개의 key값으로 정렬하는 법을 안다면 쉽게풀 수 있는 문제였다.

import sys

N = int(input())
scores = []
for i in range(N):
    scores.append(list(map(str,sys.stdin.readline().strip().split())))
    scores[i][1] = int(scores[i][1])
    scores[i][2] = int(scores[i][2])
    scores[i][3] = int(scores[i][3])

입력이 문자열과 숫자의 조합이기 때문에 모두 문자열로 입력 받은 뒤 숫자인 부분은 형변환을 해줬다.

scores.sort(key= lambda x: (-x[1],x[2],-x[3],x[0]))

for i in range(N):
    print(scores[i][0])

key로 조건에 맞게 넘겨주면 된다.

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

이름은 리스트 0번 국어는 1번 영어는 2번 수학은 3번에 입력되어있다.

국어는 내림차순이기 때문에 -x[1]
영어는 오름차순이기 때문에  x[2]
수학은 내림차순이기 때문에 -x[3]
점수가 같으면 이름이 사전 순으로 증가하기 때문에 x[0]으로 표현할 수 있다.

출력은 이름만 출력하기 때문에 정렬된 리스트의 0번만 출력하면 된다.

 

전체 코드

import sys

N = int(input())
scores = []
for i in range(N):
    scores.append(list(map(str,sys.stdin.readline().strip().split())))
    scores[i][1] = int(scores[i][1])
    scores[i][2] = int(scores[i][2])
    scores[i][3] = int(scores[i][3])



scores.sort(key= lambda x: (-x[1],x[2],-x[3],x[0]))

for i in range(N):
    print(scores[i][0])

 

만약 이름이 역순으로 정렬하는 경우라면 문자열에 대해서는 -로 연산할 수 없기 때문에 조금 더 어려웠을 것 같다.
역순으로 정렬해야 하는 경우에는 문자열을 reverse로 먼저 정렬을 한 뒤 점수에 해당하는 부분을 정렬하면 된다.

예시

scores.sort(key= lambda x: x[0], reverse = True)
scores.sort(key= lambda x: (-x[1],x[2],-x[3]))