[프로그래머스][JAVA][Lv. 1] 성격 유형 검사

문제 설명

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

---

제한 사항

---

입출력 예

---

풀이

import java.util.stream.*;
import java.util.HashMap;
import java.util.Arrays;

class Solution {
    public String solution(String[] survey, int[] choices) {
        String[] answer = new String[] {"RT", "CF", "JM", "AN"};
        int[] score = new int[] { 3, 2, 1, 0, 1, 2, 3 };
        HashMap<Character, Integer> surveyMap = new HashMap<>();
        
        for (int i = 0; i < survey.length; i++) {
            int select = 0;
            if (choices[i] <= 3) select = 0;
            else if (choices[i] >= 5) select = 1;
            else continue;
            Character c = survey[i].charAt(select);
            if (surveyMap.containsKey(c))
                surveyMap.put(c, surveyMap.get(c) + score[choices[i] - 1]);
            else surveyMap.put(c, score[choices[i] - 1]);
        }
        
        return Arrays.stream(answer)
            .map(i -> surveyMap.getOrDefault(i.charAt(0), 0) >= surveyMap.getOrDefault(i.charAt(1), 0) 
                ? String.valueOf(i.charAt(0)) : String.valueOf(i.charAt(1)))
            .collect(Collectors.joining());
    }
}

---

후기

이게 왜 4점..? 코드 확인 여러번 하고 처음 제출했을 때 바로 맞았는데 4점 받았다. 4점짜리 문제는 아닌거 같은데... 점수 책정 기준이 어떻게 되는지 찾아봐야겠다. 그나저나 요즘 스트림 공부를 하고 있는데 뭔가 재밌다. 재밌어서 문제 풀 때 자꾸 스트림을 사용하고 싶어진다. ㅋㅋ