본문 바로가기
알고리즘/프로그래머스

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

by hxxyeoniii 2024. 10. 9.

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

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


문제 풀이

groupList에 성격유형을 순서대로 기록해놓고, scoreMap에 각 알파벳과 점수가(choices를 기반으로 한) 저장해두었다.
마지막으로 groupList를 순회하며 둘 중 더 높은 점수를 가진 알파벳을 뽑아낼 수 있도록 삼항연산자를 사용하였다.
그리고 점수가 같을 경우에는 알파벳 순서대로 앞에 있는 알파벳을 뽑아내었다.
ex) "RT"일경우 scoreMap에 저장되어 있는 R, T 중에 높은 점수를 가진 알파벳을 뽑아 answer에 더해준다.

import java.util.*;

class Solution {
    public String solution(String[] surveys, int[] choices) {
        Map<Character, Integer> scoreMap = new HashMap<>();
        String[] groupList = {"RT", "CF", "JM", "AN"};
        String answer = "";

        for(int i=0; i<surveys.length; i++) {
            String survey = surveys[i];
            int choice = choices[i];
            
            char str = ' ';
            int score = 0;
            
            if(choice > 4) {
                // 동의
                str = survey.charAt(1); // 두번째 글자
                
                if(choice == 5) {
                    score = 1;
                } else if(choice == 6) {
                    score = 2;
                } else {
                    score = 3;
                }
                
            } else if(choice < 4) {
                // 비동의
                str = survey.charAt(0); // 첫번째 글자
                
                if(choice == 3) {
                    score = 1;
                } else if(choice == 2) {
                    score = 2;
                } else {
                    score = 3;
                }
            }
            
            if(scoreMap.containsKey(str)) {
                scoreMap.put(str, scoreMap.get(str) + score);
            } else {
                scoreMap.put(str, score);
            }
        }
        
        for(int i=0; i<4; i++) {
            char c1 = groupList[i].charAt(0);
            Integer a = 0;
            if(scoreMap.containsKey(c1)) {
                a = scoreMap.get(c1);
            }
            char c2 = groupList[i].charAt(1);
            Integer b = 0;
            if(scoreMap.containsKey(c2)) {
                b = scoreMap.get(c2);
            }
            
            answer += (a > b) ? c1 
                      : (a == b) ? (c1 < c2 ? c1 : c2)
                      : c2;
        }
        
        return answer;
    }
}