[c++] 프로그래머스 :: 모의고사 (brute force)

 

https://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;
    int size = answers.size();
    int correct[3] = { 0,0,0 };

    int secondList[4] = { 1,3,4,5 };
    int thirdList[5] = { 3,1,2,4,5 };

    for (int i = 0; i<size; i++) {
        int ans = answers[i];
        if ((i % 5) + 1 == ans) // 1번
            correct[0]++;
        if (i % 2 == 0 && ans == 2) // 2번
            correct[1]++;
        else if (i % 2 != 0 && secondList[(i / 2) % 4] == ans)
            correct[1]++;
        if (thirdList[(i / 2) % 5] == ans) // 3번
            correct[2]++;
    }

    if (correct[0]>correct[1]) {
        if (correct[1] >= correct[2]) {
            answer.push_back(1);
        }
        else {
            if (correct[2]>correct[0])
                answer.push_back(3);
            else if (correct[2] == correct[0]) {
                answer.push_back(1);
                answer.push_back(3);
            }
            else {
                answer.push_back(1);
            }
        }
    }
    else if (correct[0] == correct[1]) {
        if (correct[0]>correct[2]) {
            answer.push_back(1);
            answer.push_back(2);
        }
        else if (correct[0] == correct[2]) {
            answer.push_back(1);
            answer.push_back(2);
            answer.push_back(3);
        }
        else {
            answer.push_back(3);
        }
    }
    else {
        if (correct[1]<correct[2]) {
            answer.push_back(3);
        }
        else if (correct[1]>correct[2])
            answer.push_back(2);
        else {
            answer.push_back(2);
            answer.push_back(3);
        }
    }

    return answer;
}

 

발전시킬 점

+) max_element라는 함수를 쓰면 깔끔함

<algorithm> header file에 있으며 (시작주소, 끝주소)를 넘기면 max 값을 찾아서 넘겨줌

+) 패턴을 여러개를 배열에 한꺼번에 넣고 하면 깔끔함