[백준] 10825번 국영수 [Python] - 정렬

세 과목 점수와 이름 순서대로 정렬하는 문제

문제

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

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

입력

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

출력

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

 

sorted 로 정렬하고 key 파라미터에 lambda 함수로 인덱스별 우선순위를 정해서 풀었습니다.

import sys
input = sys.stdin.readline

N = int(input())
li = [[] for _ in range(N)]
for i in range(N):
    li[i] = list(input().split())
# 국어: 내림차순, 영어: 오름차순, 수학: 내림차순, 이름: 오름차순
li = sorted(li, key=lambda x:(-int(x[1]), int(x[2]), -int(x[3]), x[0]))
for n in li:
    print(n[0])  # 이름만 출력

입력 받은 정보를 바로 리스트에 저장하지 않고, 튜플로 받아온 후 점수들은 int로 변환해주고 저장했더니 메모리와 시간이 더 조금 소요됐습니다. 

import sys
input = sys.stdin.readline

N = int(input())
li = [[] for _ in range(N)]
for i in range(N):
    name, K, E, M = list(input().split())
    li[i] = name, int(K), int(E), int(M)
li.sort(key=lambda x:(-x[1], x[2], -x[3], x[0]))
for n in li:
    print(n[0])

 

정렬 함수 : sort, sorted

1. sort:
Python의 리스트(list)에서 제공하는 내장 메서드 중 하나로, 리스트의 요소들을 기본적으로 오름차순으로 정렬합니다. 이는 리스트의 순서를 변경하는 'in-place' 메서드입니다. 

numbers = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
numbers.sort()
print(numbers)  # 출력: [1, 1, 2, 3, 3, 4, 5, 5, 5, 6, 9

in-place 메서드 : 원본 변경

2. sorted:
이 함수는 `sort` 메서드와 유사하지만, 리스트의 순서를 변경하는 것이 아니라 정렬된 새로운 리스트를 반환합니다. 이렇게 반환되는 리스트는 원래의 리스트와 다른 객체입니다.

numbers = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
sorted_numbers = sorted(numbers)
print(sorted_numbers)  # 출력: [1, 1, 2, 3, 3, 4, 5, 5, 5, 6, 9]
print(numbers)         # 출력: [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]

위의 두 함수 모두 기본으로 오름차순으로 정렬하며 reverse = True로 파라미터를 변경하면 내림차순으로 정렬한다.

lamda 표현식
3. lambda:
람다(lambda) 함수는 익명 함수로, 한 줄로 표현되는 간단한 함수를 생성할 때 사용됩니다. 람다 함수는 이름 없이 정의되며, 함수의 인자와 식을 지정하여 사용합니다.

# 일반 함수 정의
def add(x, y):
    return x + y

# 람다 함수 정의
add_lambda = lambda x, y: x + y

print(add(3, 5))        # 출력: 8
print(add_lambda(3, 5)) # 출력: 8

람다 함수는 주로 간단한 연산이나, 다른 함수의 인자로 전달되는 콜백 함수 등에서 사용됩니다.

# 이름, 나이, 키
people = [
    ("Alice", 25, 160),
    ("Bob", 30, 175),
    ("Charlie", 22, 180),
    ("David", 27, 168),
    ("Eva", 24, 155)
]

# 나이 오름차순
# 키 내림차순
# 이름 알파벳순
sorted_people = sorted(people, key=lambda x: (x[1], -x[2], x[0]))

for person in sorted_people:
    print(person)

x[1]은 나이를, -x[2]는 키를 내림차순으로 정렬하기 위해 사용되었습니다. 마지막으로 x[0]는 이름을 알파벳순으로 정렬하기 위한 키로 사용됩니다.

 >>>

('Charlie', 22, 180)
('Eva', 24, 155)
('Alice', 25, 160)
('David', 27, 168)
('Bob', 30, 175)

 

 

 

예제 입력 1 

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

예제 출력 1 

Donghyuk
Sangkeun
Sunyoung
nsj
Wonseob
Sanghyun
Sei
Kangsoo
Haebin
Junkyu
Soong
Taewhan