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

 

 

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

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

 

 

배열을 세가지 미리 세팅해놓고

 

인덱스를 i%(배열의크기)로 반복하면서 answer과 맞는치 체크한다

 

 

 

Code

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

using namespace std;

vector<int> solution(vector<int> answers) {
    vector<int> answer;
    int n1[5] = {1, 2, 3, 4, 5}, 
    n2[8] = {2, 1, 2, 3, 2, 4, 2, 5}, 
    n3[10] = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};
    int sum[3] = {0,};
    
    for(int i=0; i<answers.size(); i++) {
        int ans = answers[i];
        if(ans == n1[i%5]) sum[0]++;
        if(ans == n2[i%8]) sum[1]++;
        if(ans == n3[i%10]) sum[2]++;
    }
    
    int max = *max_element(sum, sum+3);
    
    for(int i=0; i<3; i++) {
        if(sum[i] == max) answer.push_back(i+1);
    }
    
    return answer;
}