https://www.acmicpc.net/problem/10825
알고리즘 분류
- 정렬
문제
도현이네 반 학생 N명의 이름과 국어, 영어, 수학 점수가 주어진다. 이때, 다음과 같은 조건으로 학생의 성적을 정렬하는 프로그램을 작성하시오.
1. 국어 점수가 감소하는 순서로
2. 국어 점수가 같으면 영어 점수가 증가하는 순서로
3. 국어 점수와 영어 점수가 같으면 수학 점수가 감소하는 순서로
4. 모든 점수가 같으면 이름이 사전 순으로 증가하는 순서로 (단, 아스키 코드에서 대문자는 소문자보다 작으므로 사전순으로 앞에 온다.)
입력
첫째 줄에 도현이네 반의 학생의 수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 한 줄에 하나씩 각 학생의 이름, 국어, 영어, 수학 점수가 공백으로 구분해 주어진다. 점수는 1보다 크거나 같고, 100보다 작거나 같은 자연수이다. 이름은 알파벳 대소문자로 이루어진 문자열이고, 길이는 10자리를 넘지 않는다.
출력
문제에 나와있는 정렬 기준으로 정렬한 후 첫째 줄부터 N개의 줄에 걸쳐 각 학생의 이름을 출력한다.
예시
풀이
입력을 받아서 정렬한다. 이 때, 국어 점수가 높은 순서대로, 영어 점수는 작은 순서대로, 수학점수는 높은 순서부터, 모든 점수가 동일하다면 이름을 사전순으로 정렬하게 한다.
그러므로 여기서는 sort(key=lambda (-국어점수,영어점수,-수학점수,이름))을 이용한다.
소스코드
import sys
class Info:
def __init__(self, name, kor, eng, mat):
self.name = name
self.kor = int(kor)
self.eng = int(eng)
self.mat = int(mat)
students = []
for _ in range(int(sys.stdin.readline())):
name, kor, eng, mat = sys.stdin.readline().split()
student = Info(name, kor, eng, mat)
students.append(student)
students.sort(key=lambda student: (-student.kor, student.eng, -student.mat, student.name))
for student in students:
print(student.name)
1. 입력을 여러 번 받을 것이므로 sys.stdin.readline()을 사용할 것이다. 이를 위해 import sys를 하자.
2. 나는 이 자료를 딕셔너리로 정리할 생각이었기 때문에, 위에 먼저 Info class를 정의하였다. 지금 생각해보니 리스트로 받은 후 정렬할 때 split해서 튜플들을 정렬해도 될거 같다.
3. 입력을 받아 딕셔너리를 리스트에 모두 저장하고, 정렬한다. 이 때 키는, key=lambda student: (-student.kor, student.eng, -student.mat, student.name) 이다.
4. 원하는 내용만 출력한다.
'Koala - 11기 > 기초 알고리즘 스터디' 카테고리의 다른 글
[백준/python3] 2309번: 일곱 난쟁이 (0) | 2023.07.30 |
---|---|
[백준/Python] 1371번: 가장 많은 글자 (0) | 2023.07.30 |
[백준/C++] 2947번: 나무 조각 (0) | 2023.07.28 |
[백준/Python] 12719번: 비밀번호 찾기 (0) | 2023.07.23 |
[백준/Python] 7510번: 고급 수학 (0) | 2023.07.23 |