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

문제

모의고사 : 문제 링크

문제 분석

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

• 벡터내의 최대값을 찾기위해 max_element() 함수를 사용하므로 algorithm 헤더를 include
• 가장 많은 문제를 맞힌 사람을 저장할 정수형 벡터 answer, 수포자 1번부터 3번까지 맞춘 횟수를 저장할 check를 3의 크기로 0으로 초기화. 수포자 1번부터 3번까지의 찍는 방식중 반복되는 정수 배열을 supo_1부터 supo_3에 저장.
• for loop를 사용하여 배열 answers의 첫번째부터 마지막 원소까지 순회하고, if문 3개를 사용하여 answers의 현재 인덱스 원소와 supo_1에서 supo_3의 원소가 동일하다면 check의 각 원소를 1씩 늘림.(supo_1에서 supo_3의 인덱스가 초과되는 것을 방지하기 위해 현재 인덱스 i에서 각 배열 크기를 나눈 나머지를 인덱스로 설정)
• loop 탈출 후, max_element() 함수를 사용하여 정수형 변수 max_s에 check내의 원소중 가장 큰 값을 저장. 또 다른 for loop를 통해 check의 첫번째부터 마지막 원소까지 순회하고, max_s와 check의 현재 인덱스 원소가 같다면 인덱스에 1을 더한 값을 answer에 저장. loop 탈출 후, 최종적으로 저장된 answer을 return

algorithm 헤더의 max_element() 함수 사용법
*max_element(v.begin(), v.end()); => 벡터 v에서 최대값을 return
max_element(v.begin(), v.end()) - v.begin(); => 벡터 v에서 최대값의 인덱스를 return

풀이

#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> answers) {
    vector<int> answer, check(3, 0);
    vector<int> supo_1 = {1, 2, 3, 4, 5};
    vector<int> supo_2 = {2, 1, 2, 3, 2, 4, 2, 5};
    vector<int> supo_3 = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};
    
    for(int i = 0; i < answers.size(); ++i) {
        if(supo_1[i % supo_1.size()] == answers[i]) check[0]++;
        if(supo_2[i % supo_2.size()] == answers[i]) check[1]++;
        if(supo_3[i % supo_3.size()] == answers[i]) check[2]++;
    }
    int max_s = *max_element(check.begin(), check.end());
    for(int i = 0; i <= 2; ++i) {
        if(max_s == check[i]) answer.push_back(i + 1);
    }
    return answer;
}