[프로그래머스] 모의고사

https://programmers.co.kr/learn/courses/30/lessons/42840

코딩테스트 연습 - 모의고사

< 수포자1 , 맞은 점수 > , < 수포자2 , 맞은 점수 > , < 수포자3, 맞은 점수 >  와 같은 레코드를 가지는

해쉬 테이블을 만들어서 해결할 수 있다.

수포자1, 2, 3 이 문제를 찍는 방식은 각각 5, 8, 10 의 순환마디를 가지고 있으므로 이에 따라서

각각 맞은 점수를 증가시켜준다.

그리고 해쉬 테이블을 key 가 아닌 value 값을 기준으로 sort 해주면 편한데

이를 구현하기 위해서 해쉬 테이블의 record 를 원소로 가지는 vector 을 선언한 뒤 sorting 을 해준다.

vector 컨테이너는 메모리가 연속적이기 때문이다.

 

정렬을 마치면 v[0], v[1], v[2] 는 각각 고득점한 순서로 정렬된다. (compare 함수 주의)

1등이 몇명인지에 따라 answer 에 누구를 insert 하고 정렬을 할지 말지를 결정한다.

 

v[0].second 가 v[1] 보다 크면 단독 1등이 발생한 것이므로 v[0].first 만 push 해주고 끝낸다.

 

v[0].second 가 v[1].second 와 같고 v[1].second 과 v[2].second 가 다르다면 1등이 2명 발생한 것이므로,

v[0].first 와 v[1].first 를 push 한 뒤에 정렬하고 끝낸다.

 

v[0].second 와 v[1].second 와 v[2].second 가 모두 같다면 1등이 3명 발생한 것이므로 각각의 first 를 push 하고

정렬한 뒤 끝낸다.

 

 

 

[C++]

#include <string>
#include <vector>
#include <unordered_map>
#include <algorithm>
#include <iostream>

using namespace std;
 
int one[] = { 1, 2, 3, 4, 5 }; // 5
int two[] = { 2, 1, 2, 3, 2, 4, 2, 5 }; // 8
int three[] = { 3, 3, 1, 1, 2, 2, 4, 4, 5, 5 }; // 10

vector<int> solution(vector<int> answers) {
    vector<int> answer;
    int num = answers.size();

    unordered_map <int, int> idx;
    idx.insert(make_pair(1, 0));
    idx.insert(make_pair(2, 0));
    idx.insert(make_pair(3, 0));

    int ans = 0;
    for (int i = 0; i < num; i++)
    {
        ans = answers[i];
        if (ans == one[i % 5])
        {
            idx[1]++;
        }
        if (ans == two[i % 8])
        {
            idx[2]++;
        }
        if (ans == three[i % 10])
        {
            idx[3]++;
        }
    }

    vector<pair<int, int>> v(idx.begin(), idx.end());

    sort(v.begin(), v.end(), [](pair<int, int> a, pair<int, int> b)
        {
            return a.second > b.second;
        });


    if (v[0].second > v[1].second)
    {
        answer.push_back(v[0].first);
        return answer;
    }
    if (v[0].second == v[1].second && v[1].second != v[2].second)
    {
        answer.push_back(v[0].first);
        answer.push_back(v[1].first);
        sort(answer.begin(), answer.end());
        return answer;
    }
    if (v[0].second == v[1].second && v[1].second == v[2].second)
    {
        answer.push_back(v[0].first);
        answer.push_back(v[1].first);
        answer.push_back(v[2].first);
        sort(answer.begin(), answer.end());
    }
    return answer;
}