[백준] 10825번 국영수
in Algorithms on BOJ
std::sort - comp function
문제설명
문제
도현이네 반 학생 N명의 이름과 국어, 영어, 수학 점수가 주어진다. 이때, 다음과 같은 조건으로 학생의 성적을 정렬하는 프로그램을 작성하시오.
국어 점수가 감소하는 순서로 국어 점수가 같으면 영어 점수가 증가하는 순서로 국어 점수와 영어 점수가 같으면 수학 점수가 감소하는 순서로 모든 점수가 같으면 이름이 사전 순으로 증가하는 순서로 (단, 아스키 코드에서 대문자는 소문자보다 작으므로 사전순으로 앞에 온다.)
입력
첫째 줄에 도현이네 반의 학생의 수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 한 줄에 하나씩 각 학생의 이름, 국어, 영어, 수학 점수가 공백으로 구분해 주어진다. 점수는 1보다 크거나 같고, 100보다 작거나 같은 자연수이다. 이름은 알파벳 대소문자로 이루어진 문자열이고, 길이는 10자리를 넘지 않는다.
출력
문제에 나와있는 정렬 기준으로 정렬한 후 첫째 줄부터 N개의 줄에 걸쳐 각 학생의 이름을 출력한다.
접근
이 문제는 std::sort함수의 comp함수에 대한 문제이다. std::sort에서 comp함수만 조건에맞게 새로 정의해주면 쉽게 풀린다. 아래의 cpp-reference를 확인해보면 세번째 인자로 넘어가는 comp함수는 두 매개변수 중 앞 매개변수가 더 작을때(앞에 나올때) true를 리턴한다.(더 크거나 뒤에 나올때는 false 리턴)
이를 바탕으로 문제 조건에 맞는 comp함수를 작성해서 sort의 세번째 인자로 넘겨주면 된다.
코드
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
struct student
{
string name;
int kor, eng, math;
};
bool cmp(student a, student b){
if(a.kor > b.kor)
return true;
else if(a.kor == b.kor){
if(a.eng < b.eng)
return true;
else if(a.eng == b.eng){
if(a.math > b.math)
return true;
else if(a.math == b.math){
if(a.name < b.name)
return true;
}
}
}
return false;
}
int main(){
cin.tie(0);
cout.tie(0);
ios_base::sync_with_stdio(false);
int n;
cin >> n;
vector<student> v;
for(int i=0; i<n; i++){
student tmp;
cin >> tmp.name >> tmp.kor >> tmp.eng >> tmp.math;
v.push_back(tmp);
}
sort(v.begin(), v.end(), cmp);
for(auto s: v){
cout << s.name << '\n';
}
}
참고
BOJ #10825
cpp reference - std::sort
이 글이 도움이 되셨다면 추천 클릭을 부탁드립니다 :)
