BOJ 10825 (국영수)

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

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

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

#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
int studentCount;

void fast_io() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);
}

class Student {
public:
	string name;
	int koreanScore;
	int englishScore;
	int mathScore;
};


void input(Student* st) {
	cin >> studentCount;
	for (int i = 0; i < studentCount; i++) {
		cin >> st[i].name >> st[i].koreanScore
			>> st[i].englishScore >> st[i].mathScore;
	}
}
bool comp(Student s1, Student s2) {
	if (s1.koreanScore == s2.koreanScore &&
		s1.englishScore == s2.englishScore &&
		s1.mathScore == s2.mathScore)
		return s1.name > s2.name;

	else if (s1.koreanScore == s2.koreanScore &&
		s1.englishScore == s2.englishScore)
		return s1.mathScore < s2.mathScore;

	else if (s1.koreanScore == s2.koreanScore)
		return s1.englishScore > s2.englishScore;

	else
		return s1.koreanScore < s2.koreanScore;
}

int main() {
	fast_io();
	Student* studentInfo = new Student[100001];
	input(studentInfo);
	sort(studentInfo, studentInfo + studentCount, comp);
	for (int i = studentCount - 1; i >= 0; i--) {
		cout << studentInfo[i].name << '\n';
	}
	return 0;
}

   학생 수가 최대 10만이라 O(N^2) 이상의 시간이 걸리는 알고리즘으로 문제를 풀면 시간초과가 날 것이다. 맨 처음에 어떻게 입력을 받을까 고민하다가 학생 이름과 점수가 하나로 묶여야 하기 때문에 Class를 설계했다.
오랜만에 Class를 만들어 보니 매우 어색하고 문법이 아리까리 했다....

문제의 핵심은 O(NlogN)이 걸리는 sort 알고리즘의 3번째 인자를 잘 구현하는 것이다.
오름차순 내림차순을 반대로 하여 거꾸로 출력을 시켰는데 등호 방향을 모두 반대로 바꿔준다면 0번째 index부터 원하는 답과 일치할 것 같다.

comp함수를 구성할 때 if ~ else if ~ else 조건문에서 반복되는 코드가 있다 Clean Code 책에서 반복을 피하라고 했으므로 이를 3중첩 if문으로 구현한다면 분기는 깊어지지만 반복을 피할 수 있을 것 같다.

시간 복잡도 : O(NlogN)

숏코딩1 -> Class를 설계하지 않는 풀이

#include<bits/stdc++.h>
using namespace std;
vector<tuple<int,int,int,string>> v;
int main(){
    int n; cin>>n;
    int k,e,m; string s;
    for(int i=0;i<n;i++){
        cin>>s>>k>>e>>m;
        v.push_back({-k,e,-m,s});
    }
    sort(v.begin(),v.end());
    for(int i=0;i<n;i++) cout<<get<3>(v[i])<<'\n';
}

tuple 자료형을 이용하고 내림 차순이 필요한 부분에서는 -1을 곱해 sort함수의 3번째 인자인 comp 함수를 사용하지않고도 사용한 풀이 인것 같다.
이름은 tuple 자료형의 get<index>를 통해 가져와 출력하고 있다.

숏코딩 2

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main()
{
    int b, c, e, f, i;
    string d;
    vector<pair<int, string>>a;
    cin>>b;
    for(i=0 ; i<b ; i++)
    {
        cin>>d>>c>>e>>f;
        a.push_back({-10000*c+100*e-f, d});
    }
    sort(a.begin(), a.end());
    for(auto &x:a)
    {
        cout<<x.second<<'\n';
    }
}

이건 아예 국영수 점수를 계산을 통해 분리한 것 같다.

처음엔 어떻게 입력을 받을까 많이 고민했는데 Class 설계하지 않고 푸는 법을 보니 놀랐다. 안드로이드를 공부하고 있어 C++로 코딩테스트 준비 외에는 코드작성할 일이 없겠지만 문법 공부를 가끔씩 해줘야겠다.