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

문제 : programmers.co.kr/learn/courses/30/lessons/42840

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

 

#include <string>
#include <vector>
#include <algorithm>
 
using namespace std;
 
vector<int> solution(vector<int> answers) {
    vector<int> answer;
    vector<int> a_arr = { 1,2,3,4,5 };
    vector<int> b_arr = { 2,1,2,3,2,4,2,5 };
    vector<int> c_arr = { 3,3,1,1,2,2,4,4,5,5 };
 
    vector<int> score(3);
 
    for (int i = 0; i < answers.size(); i++) {
        if (answers[i] == a_arr[i % 5]) {
            score[0]++;
        }
        if (answers[i] == b_arr[i % 8]) {
            score[1]++;
        }
        if (answers[i] == c_arr[i % 10]) {
            score[2]++;
        }
    }
 
    int max_score = *max_element(score.begin(),score.end()); 
    for (int i = 0; i < 3; i++) {
        if (score[i] == max_score) {
            answer.push_back(i + 1);
        }
    } // 최대값 찾기
    return answer;
}