[프로그래머스/C++] 성격 유형 검사하기

1. 개요

https://school.programmers.co.kr/learn/courses/30/lessons/118666

프로그래머스

1-1. 설명

카카오 성격 유형 검사지를 만드려고 한다. 각 설문지와 점수가 주어졌을때, 나올수 있는 결과를 출력하는 프로그램을 구현한다. 각 성격 유형과 선택지의 자세한 설명은 위의 링크를 통해 참조하면 된다. 

1-2. 제한 사항

 - survey의 길이는 1이상 1,000 이하이며, 원소는 8개 중 택 1로 나옴

 - choices의 길이는 survey 길이와 동일하며, 원소는 1이상 7이하 정수

2. 구현

2-1. 풀이

문제에 적혀있는대로, 잘 구현하면 정답이 나온다. map으로 할지, 구조체를 통해 여러 변수를 만들지 고민하다가 후자를 선택했다. map을 통해 하나의 변수로 두어 구현하면 조금 더 코드가 깔끔하고 최적화될 것이라 생각한다. 코드가 난잡하여, 시간이 된다면 조금 더 간추려봐야겠다. 코드를 참고하려면 내 코드보다는 다른사람의 코드를 참고하는 것이 더 깔끔하다고 생각한다. 

2-2. 코드

#include <string>
#include <vector>

using namespace std;

struct TYPE {
    char type;
    int num;
};

TYPE Type1[2] = {{'R', 0}, {'T', 0}}; // 라이언, 튜브
TYPE Type2[2] = {{'C', 0}, {'F', 0}}; // 콘, 프로도
TYPE Type3[2] = {{'J', 0}, {'M', 0}}; // 제이지, 무지
TYPE Type4[2] = {{'A', 0}, {'N', 0}}; // 어피치, 네오
// 1 ~ 3 , 5 ~ 7
string solution(vector<string> survey, vector<int> choices) {
    string answer = "";
    for (int i = 0; i < survey.size(); i++) {
        if (choices[i] == 4) continue;
        else if (choices[i] == 1) choices[i] = 3;
        else if (choices[i] == 3) choices[i] = 1;
        
        if (survey[i] == "RT" || survey[i] == "TR") { // type 1
            if (choices[i] >= 1 && choices[i] <= 3) { // 3 ~ 1 
                for (int j = 0; j < 2;j++) {
                    if (Type1[j].type == survey[i][0]) Type1[j].num += choices[i];
                }
            }
            else if (choices[i] >= 5 && choices[i] <= 7){ // 1 ~ 3 -> choices[i] - 4
                for (int j = 0; j < 2; j++) {
                    if (Type1[j].type == survey[i][1]) Type1[j].num += (choices[i] - 4);
                }
            }
            
        }
        else if (survey[i] == "CF" || survey[i] == "FC") { // type 2
            if (choices[i] >= 1 && choices[i] <= 3) { // 3 ~ 1 
                for (int j = 0; j < 2;j++) {
                    if (Type2[j].type == survey[i][0]) Type2[j].num += choices[i];
                }
            }
            else if (choices[i] >= 5 && choices[i] <= 7){ // 1 ~ 3 -> choices[i] - 4
                for (int j = 0; j < 2; j++) {
                    if (Type2[j].type == survey[i][1]) Type2[j].num += (choices[i] - 4);
                }
            }
        }
        else if (survey[i] == "JM" || survey[i] == "MJ") { // type 3
            if (choices[i] >= 1 && choices[i] <= 3) { // 3 ~ 1 
                for (int j = 0; j < 2;j++) {
                    if (Type3[j].type == survey[i][0]) Type3[j].num += choices[i];
                }
            }
            else if (choices[i] >= 5 && choices[i] <= 7){ // 1 ~ 3 -> choices[i] - 4
                for (int j = 0; j < 2; j++) {
                    if (Type3[j].type == survey[i][1]) Type3[j].num += (choices[i] - 4);
                }
            }

        }
        else if (survey[i] == "AN" || survey[i] == "NA"){ // type 4
            if (choices[i] >= 1 && choices[i] <= 3) { // 3 ~ 1 
                for (int j = 0; j < 2;j++) {
                    if (Type4[j].type == survey[i][0]) Type4[j].num += choices[i];
                }
            }
            else if (choices[i] >= 5 && choices[i] <= 7){ // 1 ~ 3 -> choices[i] - 4
                for (int j = 0; j < 2; j++) {
                    if (Type4[j].type == survey[i][1]) Type4[j].num += (choices[i] - 4);
                }
            }
        }
    }
    if (Type1[0].num >= Type1[1].num) answer += Type1[0].type;
    else answer += Type1[1].type;
    
    if (Type2[0].num >= Type2[1].num) answer += Type2[0].type;
    else answer += Type2[1].type;
    
    if (Type3[0].num >= Type3[1].num) answer += Type3[0].type;
    else answer += Type3[1].type;
    
    if (Type4[0].num >= Type4[1].num) answer += Type4[0].type;
    else answer += Type4[1].type;
    
    return answer;
}