[프로그래머스/C++ 문제 풀이] Lv. 1 - 모의고사

문제 설명

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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하는 값을 오름차순 정렬해주세요.

입출력 예

 

입출력 예 #1

• 수포자 1은 모든 문제를 맞혔습니다.
• 수포자 2는 모든 문제를 틀렸습니다.
• 수포자 3은 모든 문제를 틀렸습니다.

따라서 가장 문제를 많이 맞힌 사람은 수포자 1입니다.

 

입출력 예 #2

• 모든 사람이 2문제씩을 맞췄습니다.

---

문제 풀이

이번 문제는 3명의 학생이 정해진 규칙대로 문제를 찍었을 때 가장 많이 맞춘 학생을 찾는 문제입니다.

 

저의 경우 queue에 개별 학생의 답안지를 넣고, 답안(answer)을 순회하며 queue의 front() 값과 비교하여 카운트를 증가시켰으며, 이후 가장 앞 요소를 가장 뒤로 다시 넣어주는 방식으로 문제를 해결하였습니다.

 

다 풀고 생각해보니.. 다음과 같이 간단히 answer 배열을 순회하며 [i % 학생 답안 수]를 통해 굳이 큐를 사용하지 않고도 반복시킬 수 있겠다는 생각이 들었습니다.

for(int i=0; i<answers.size(); i++) 
{
    if(answers[i] == v1[i % v1.size()]) cnt[0]++;
    if(answers[i] == v2[i % v2.size()]) cnt[1]++;
    if(answers[i] == v3[i % v3.size()]) cnt[2]++;
}

정답 코드

풀이 시간 : 13m 47s

#include <string>
#include <vector>
#include <queue>

using namespace std;

// 맞춘 횟수 저장용 배열
int cnt[3];

vector<int> solution(vector<int> answers) {
    vector<int> answer;
    
    // 1번 : 1 2 3 4 5
    vector<int> v1 = {1, 2, 3, 4, 5};
    queue<int> q1;
    
    for (int i = 0; i < v1.size(); i++)
    {
        q1.push(v1[i]);
    }
    

    // 2번 : 2 1 2 3 2 4 2 5
    vector<int> v2 = {2, 1, 2, 3, 2, 4, 2, 5};
    queue<int> q2;
    
    for (int i = 0; i < v2.size(); i++)
    {
        q2.push(v2[i]);
    }

    // 3번 : 3 3 1 1 2 2 4 4 5 5
    vector<int> v3 = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};
    queue<int> q3;
    
    for (int i = 0; i < v3.size(); i++)
    {
        q3.push(v3[i]);
    }
    
    // 답안 순회
    for (int num : answers)
    {
        // 학생이 답을 맞춘 경우 카운트 증가
        if (q1.front() == num) cnt[0]++;
        if (q2.front() == num) cnt[1]++;
        if (q3.front() == num) cnt[2]++;
        
        // 하나씩 추출하여 큐의 맨 뒤로 보내기
        int temp = 0;
        
        temp = q1.front();
        q1.pop();
        q1.push(temp);
        
        temp = q2.front();
        q2.pop();
        q2.push(temp);
        
        temp = q3.front();
        q3.pop();
        q3.push(temp);
    }
    
    // 최대 값 구하기
    int mx = max(cnt[0], max(cnt[1], cnt[2]));
    
    // 최대 값인 학생들 추가
    for (int i = 0; i < 3; i++)
    {
        if (cnt[i] == mx)
        {
            answer.push_back(i + 1);
        }
    }
    
    
    return answer;
}