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

프로그래머스

완전탐색

모의고사

 

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

코딩테스트 연습 - 모의고사 | 프로그래머스

 

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> answers) {
    vector<int> answer;
    int su1[5] = {1,2,3,4,5};
    int su2[8] = {2,1,2,3,2,4,2,5};
    int su3[10] = {3,3,1,1,2,2,4,4,5,5};
    int cnt1=0, cnt2=0, cnt3=0, max;
    int size = answers.size();
    
    for(int i=0; i<size; i++) {
        if(answers[i] == su1[i%5]) {
            cnt1++;
        }
        if(answers[i] == su2[i%8]) {
            cnt2++;
        }
        if(answers[i] == su3[i%10]) {
            cnt3++;
        }
    }
    
    max = cnt1;
    answer.push_back(1);
    if(cnt1 == cnt2) {
        answer.push_back(2);
    }
    else if(cnt1 < cnt2) {
        answer.clear();
        answer.push_back(2);
        max = cnt2;
    }
    
    if(max == cnt3) {
        answer.push_back(3);
    }
    else if(max < cnt3) {
        answer.clear();
        answer.push_back(3);
        max = cnt3;
    }
    
    return answer;
}