[백준/c++] 10825: 국영수

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

10825번: 국영수

문제

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

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

입력

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

출력

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

풀이

1. 힌 학생 당 다양한 정보(국어, 영어, 수학, 이름)를 저장해야 하므로 구조체를 사용했다.

2. c++에서는 sort함수의 3번째 인자를 이용해 sort를 커스텀할 수 있다. 따라서 compare이라는 함수를 만들어 문제 조건에 맞게 정렬 조건을 제시해주었다.

3. 구조체를 함수 인자로 전달할 때 포인터를 사용하지 않고 그냥 Student a 이런 식으로 할 경우 원본 데이터가 바뀌는 것이 아니기 때문에 포인터로 전달해준다.

★새롭게 알게 된 사실: c++에서 시간초과가 날 경우 endl 대신 '\n'을 사용하면 정답처리 되는 경우가 있었다. 이는 둘의 동작 방식이 다르기 때문인데 버퍼와 관련이 있다. (참고: https://yechoi.tistory.com/48)

코드

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;

struct Student {
    int kor;
    int eng;
    int math;
    string name;
};

bool compare(Student& a, Student& b) {
    if(a.kor != b.kor) return a.kor > b.kor;
    else {
        if(a.eng == b.eng) {
            if(a.math == b.math) return a.name < b.name;
            else return a.math > b.math;
        }
        else return a.eng < b.eng;
    }
}

int main() {
    int N;
    cin >> N;
    vector<Student> student_list(N);
    for(int i=0;i<N;i++)
        cin >> student_list[i].name >> student_list[i].kor >> student_list[i].eng >> student_list[i].math;
    sort(student_list.begin(), student_list.end(), compare);
    for(int i=0;i<N;i++)
        cout << student_list[i].name << "\n";
    return 0;
}