[Programmers] 118666번 - 성격 유형 검사하기 [Java][C++]

문제 링크

---

1. 아이디어

주어진 설문조사와 선택을 통해 성격 유형을 구하는 문제로 지표별로 어떤 유형이 더 우세한지를 구해주기만 하면 된다. 이를 위해 해시맵을 활용해 key에 성격 유형, value에 점수를 담은 후 각 지표별로 더 우세한 유형을 찾아서 문자열로 반환해줬다. 지표 내에서 동점 시 사전 순으로 빠른 유형을 선택해야 함에 주의해야 한다.

---

2. 복잡도

시간복잡도	공간복잡도
$O(N)$	$O(1)$

$N$ = survey 길이(= choices 길이)

---

3. 코드

풀이 [Java][C++]

각 지표 내에서 사전 순인 R, C, J, A에 대해 >=으로 처리해주었다.

import java.util.*;

class Solution {
    public String solution(String[] survey, int[] choices) {
        Map<Character, Integer> map = new HashMap<>();

        for (int i = 0; i < survey.length; i++) {
            char t1 = survey[i].charAt(0);
            char t2 = survey[i].charAt(1);
            int c = choices[i];

            if (c == 1) {
                map.put(t1, map.getOrDefault(t1, 0) + 3);
            } else if (c == 2) {
                map.put(t1, map.getOrDefault(t1, 0) + 2);
            } else if (c == 3) {
                map.put(t1, map.getOrDefault(t1, 0) + 1);
            } else if (c == 5) {
                map.put(t2, map.getOrDefault(t2, 0) + 1);
            } else if (c == 6) {
                map.put(t2, map.getOrDefault(t2, 0) + 2);
            } else if (c == 7) {
                map.put(t2, map.getOrDefault(t2, 0) + 3);
            }
        }

        String ans = "";
        if (map.getOrDefault('R', 0) >= map.getOrDefault('T', 0)) {
            ans += "R";
        } else {
            ans += "T";
        }

        if (map.getOrDefault('C', 0) >= map.getOrDefault('F', 0)) {
            ans += "C";
        } else {
            ans += "F";
        }

        if (map.getOrDefault('J', 0) >= map.getOrDefault('M', 0)) {
            ans += "J";
        } else {
            ans += "M";
        }

        if (map.getOrDefault('A', 0) >= map.getOrDefault('N', 0)) {
            ans += "A";
        } else {
            ans += "N";
        }

        return ans;
    }
}

#include <string>
#include <unordered_map>
#include <vector>

using namespace std;

string solution(vector<string> survey, vector<int> choices) {
    unordered_map<char, int> mp;

    for (int i = 0; i < survey.size(); i++) {
        char t1 = survey[i].front();
        char t2 = survey[i].back();
        int c = choices[i];

        if (c == 1) {
            mp[t1] += 3;
        } else if (c == 2) {
            mp[t1] += 2;
        } else if (c == 3) {
            mp[t1] += 1;
        } else if (c == 5) {
            mp[t2] += 1;
        } else if (c == 6) {
            mp[t2] += 2;
        } else if (c == 7) {
            mp[t2] += 3;
        }
    }

    string ans = "";
    if (mp['R'] >= mp['T']) {
        ans += "R";
    } else {
        ans += "T";
    }

    if (mp['C'] >= mp['F']) {
        ans += "C";
    } else {
        ans += "F";
    }

    if (mp['J'] >= mp['M']) {
        ans += "J";
    } else {
        ans += "M";
    }

    if (mp['A'] >= mp['N']) {
        ans += "A";
    } else {
        ans += "N";
    }

    return ans;
}

---