[자바] 성격유형 검사하기

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

프로그래머스

비교적 쉬운 문제였지만, 아래 다른 분이 stream으로 푼 풀이를 보고 이렇게 풀어도 되겠구나 하고 기록하기 위해 포스팅한다.

 

<내 풀이> 

배열에 각 문자를 저장하고, 각 문자별 점수는 map에 저장.

그리고, map의 값을 빼서 배열(c)의 문자로 담는다. 

import java.util.*;

class Solution {
    public String solution(String[] survey, int[] choices) {
        String answer ="";
        char[][] c = {{'R','T'},{'C','F'},{'J','M'},{'A','N'}};
        Map<Character, Integer> mapp = new HashMap<>(); 
        
        for (int i=0; i<survey.length; i++) {
            if (choices[i]<4) {
                char left = survey[i].charAt(0);
                mapp.put(left,mapp.getOrDefault(left, 0)+4-choices[i]);
            }
            else if (choices[i]>4) {
                char right = survey[i].charAt(1);
                mapp.put(right,mapp.getOrDefault(right,0)+choices[i]-4);
            }
        }
        for (int i=0; i<4; i++) {
            answer+= (mapp.getOrDefault(c[i][0],0)>=mapp.getOrDefault(c[i][1],0) ? c[i][0] : c[i][1]);
        }
        
        return answer;
    }
}

 

<다른 풀이>

getOrdefault()메소드와 stream을 사용해서도 풀 수 있구나라는 걸 저장하고 싶어서 기록.

stream은 알고리즘할 때 풀어보진 않았는데, stream에 조건문을 써서 이렇게 간단하게 풀 수 있었다.

import java.util.*;

class Solution {
    public String solution(String[] survey, int[] choices) {
        Map<Character, Integer> mapp = new HashMap<>(); 
        
        for (int i=0; i<survey.length; i++) {
            if (choices[i]<4) {
                char left = survey[i].charAt(0);
                mapp.put(left,mapp.getOrDefault(left, 0)+4-choices[i]);
            }
            else if (choices[i]>4) {
                char right = survey[i].charAt(1);
                mapp.put(right,mapp.getOrDefault(right,0)+choices[i]-4);
            }
        }
        return new StringBuilder()
            .append(mapp.getOrDefault('R',0) >= mapp.getOrDefault('T',0) ? 'R':'T')
            .append(mapp.getOrDefault('C',0) >= mapp.getOrDefault('F',0) ? 'C':'F')
            .append(mapp.getOrDefault('J',0) >= mapp.getOrDefault('M',0) ? 'J':'M')
            .append(mapp.getOrDefault('A',0) >= mapp.getOrDefault('N',0) ? 'A':'N')
            .toString();
    }
}