[level 1] 모의고사 - 42840

문제 설명
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.

1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...

1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한 조건
시험은 최대 10,000 문제로 구성되어있습니다.
문제의 정답은 1, 2, 3, 4, 5중 하나입니다.
가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.
입출력 예
answers return
[1,2,3,4,5] [1]
[1,3,2,4,2] [1,2,3]

 

---

설계

각 학생의 점수를 vector<vector<int>>로 해서 { (학생번호), (학생점수) }를 저장한다. 점수를 통해 내림차순 정렬을 하고 1등의 점수와 2등 점수를 비교하고 다르다면 1등만 같다면 return vector<int>에 푸시 해준다.

 

구현

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> answers) {
    vector<int> answer(3);

    vector<int> Student1 = { 1, 2, 3, 4, 5 };
    vector<int> Student2 = { 2, 1, 2, 3, 2, 4, 2, 5 };
    vector<int> Student3 = { 3, 3, 1, 1, 2, 2, 4, 4, 5, 5 };

    vector<vector<int>> Score = { {1,0}, {2,0}, {3,0} };

    for (int i = 0; i < answers.size(); i++)
    {
        if (Student1[i % 5] == answers[i])
        {
            Score[0][1]++;
        }

        if (Student2[i % 8] == answers[i])
        {
            Score[1][1]++;
        }

        if (Student3[i % 10] == answers[i])
        {
            Score[2][1]++;
        }
    }

    sort(Score.begin(), Score.end(), [](const vector<int>& a, vector<int>& b) {
        if (a[1] == b[1])
        {
            return a[0] < b[0];
        }
    
        return a[1] > b[1];
    });
   
    for (int j = 0; j < 3; j++)
    {
        if (j != 0 && Score[j-1][1] != Score[j][1])
        {
            answer.resize(j);
            return answer;
        }

        answer[j] = Score[j][0];
    }

    return answer;
}

 

---

AI 한테 리펙토링 받은 코드

#include <vector>
#include <algorithm> // std::max 를 사용하기 위해 포함

using namespace std;

vector<int> solution(vector<int> answers) {
    // 결과를 담을 빈 벡터 생성
    vector<int> answer;

    // 1. 수포자들의 찍는 패턴을 미리 정의합니다.
    vector<int> student1 = { 1, 2, 3, 4, 5 };
    vector<int> student2 = { 2, 1, 2, 3, 2, 4, 2, 5 };
    vector<int> student3 = { 3, 3, 1, 1, 2, 2, 4, 4, 5, 5 };

    // 2. 각 수포자의 맞힌 문항 수를 저장할 1차원 벡터를 선언합니다.
    // scores[0]은 1번 수포자의 점수, scores[1]은 2번 수포자의 점수가 됩니다.
    // 3명의 점수를 모두 0점(초기값)으로 세팅합니다.
    vector<int> scores(3, 0); 

    // 3. 정답 배열을 처음부터 끝까지 순회하며 채점을 진행합니다.
    for (int i = 0; i < answers.size(); i++) {
        // 숫자를 직접 적지 않고 .size()를 사용해 배열의 길이로 나눈 나머지를 구합니다.
        if (student1[i % student1.size()] == answers[i]) {
            scores[0]++;
        }
        if (student2[i % student2.size()] == answers[i]) {
            scores[1]++;
        }
        if (student3[i % student3.size()] == answers[i]) {
            scores[2]++;
        }
    }

    // 4. 가장 높은 점수를 찾습니다.
    // std::max({a, b, c}) 를 사용하면 여러 값 중 가장 큰 값을 쉽게 구할 수 있습니다.
    int max_score = max({scores[0], scores[1], scores[2]});

    // 5. 최고 점수와 같은 점수를 가진 사람을 정답(answer) 배열에 차례대로 넣습니다.
    // i가 0부터 2까지(1번 수포자부터 3번 수포자까지) 순서대로 탐색하므로, 
    // 동점자가 여러 명이어도 자연스럽게 오름차순으로 정답 배열에 들어가게 됩니다. (정렬 불필요!)
    for (int i = 0; i < 3; i++) {
        if (scores[i] == max_score) {
            // 수포자 번호는 1, 2, 3번이므로 인덱스 i에 1을 더해줍니다.
            answer.push_back(i + 1); 
        }
    }

    return answer;
}

리팩토링 포인트

1. 이중 벡터(vector<vector<int>>) 사용 지양: 단순히 점수를 저장하기 위해 2차원 벡터를 사용하면, 내부적으로 메모리 할당이 여러 번 일어나 속도가 느려집니다. 3명의 점수만 저장하면 되므로 **1차원 배열(또는 크기가 3인 1차원 벡터)**로 충분합니다.
2. 불필요한 정렬(sort) 제거: 데이터가 3개뿐이며, 인덱스 0, 1, 2가 각각 수포자 1, 2, 3번을 의미합니다. 점수 배열에서 가장 큰 값(최고점)을 찾은 뒤, 앞에서부터 차례대로 그 최고점과 같은지 비교해서 정답 배열에 넣으면 자동으로 오름차순 정렬이 됩니다. 즉, 람다 함수를 쓴 무거운 sort 과정을 아예 생략할 수 있습니다.
3. 매직 넘버(Magic Number) 제거: i % 5, i % 8처럼 숫자를 직접 쓰기보다는 student1.size()와 같이 벡터의 크기를 가져오는 함수를 쓰면, 나중에 배열 길이가 바뀌어도 코드를 수정할 필요가 없어 안전합니다.
4. 람다 매개변수 오류 수정 (참고): 작성하신 람다식 [](const vector<int>& a, vector<int>& b)에서 b 앞에 const가 빠져있습니다. 둘 다 값을 읽기만 하므로 const를 붙여주는 것이 올바른 C++ 문법입니다.

---

회고

벡터에 점수만 넣으면 해당 점수가 누구인지를 알 수 없어서 이중 백터를 사용하였는데 점수를 저장한 벡터 자체를 오름차순하는 것이 아니라 최대 점수를 다른 변수에 저장하고 최고점수와 학생의 점수 백터를 비교해 저장하면 더 효율적이라는 점을 캐치할 수 있었다.