[Java] 프로그래머스 - 성격 유형 검사하기 (1단계)

📝 문제

프로그래머스 - 성격 유형 검사하기 (1단계)

---

📝 답안

📌 작성 코드

class Solution {
    public String solution(String[] survey, int[] choices) {
        String answer = "";
        String mbti = "RTCFJMAN";
        int[] score = new int[8];
        
        for (int i = 0; i < survey.length; i++) {
            String[] word = survey[i].split("");
            if (choices[i] < 4) {
                score[mbti.indexOf(word[0])] += (4 - choices[i]);
            } else if (choices[i] > 4) {
                score[mbti.indexOf(word[1])] += (choices[i] - 4);
            }
        }
        
        for (int i = 0; i < 4; i++) {
            int a = i * 2;
            int b = i * 2 + 1;
            
            if (score[a] >= score[b]) {
                answer += mbti.substring(a, a+1);
            } else {
                answer += mbti.substring(b, b+1);
            }
        }
        
        return answer;
    }
}

---

📌 결과