[프로그래머스] LV.1 완주하지 못한 선수 (HashMap, StringBuilder, getOrDefault, Java)

 

문제설명

입력 & 출력

나의 풀이

import java.util.*;

class Solution {
    public String solution(String[] participant, String[] completion) {
        HashMap<String,Integer> map = new HashMap<>();
        StringBuilder sb = new StringBuilder();
        
        for(int i = 0 ; i < participant.length ; i++){
            String name = participant[i];
            if(map.get(name) != null){
                map.put(name, map.get(name) + 1);    
            }else{
                map.put(name,1);
            }   
        }
        
        for(int i = 0 ; i < completion.length ; i++){
            String name = completion[i];
            if(map.get(name) != null){
                map.put(name,map.get(name) - 1);
            }
        }
        
        for (String key : map.keySet()) {
            if (map.get(key) != 0) {
                sb.append(key);
            }
        }
        
        return sb.toString();
    }
}

 

이번 문제는 HashMap을 활용할 수 있는 문제입니다. 전체 참가자 배열과 완주자 배열이 주어지고, 완주하지 못한 사람의 이름을 반환하면 되는 문제입니다.

 

처음에는 HashSet을 사용하여 코드를 작성했지만 중복된 이름이 존재하는 경우 HashSet의 특성상 중복을 허용하지 않기에 적절하지 않아 HashMap으로 풀어봤습니다.

 

먼저 HashMap을 선언하고, 각 완주자를 map에 담아줍니다. 이때 HashMap은 Key를 조회할 때 키값이 없으면 NULL을 반환하기 때문에  NULL체크를 해야합니다.

 

만약 전체 참가자를 접근할 때 해당 참가자가 map에 없으면 key를 참가자의 이름으로, value를 1로 put 해줍니다.

그리고 만약 동일이름의 참가자가 존재한다면 get() 메서드로 해당 키를 가진 참가자의 값을 가져와 해당 값에 1을 더해준 값을 넣어줍니다.

 

즉 동일이름이 있다면 그 값에 1을 더해줍니다.

 

그러면 다음과 같이 HashMap이 형성되게 됩니다.

 

• ["marina", "josipa", "nikola", "vinko", "filipa"], ["josipa", "filipa", "marina", "nikola"]
  • {marina=1, filipa=1, nikola=1, josipa=1, vinko=1}
• ["mislav", "stanko", "mislav", "ana"], ["stanko", "ana", "mislav"]
  • {ana=1, mislav=2, stanko=1}

---

그리고 문제의 목적은 완주하지 못한 참가자를 찾는 것 이기 때문에 참가자 배열인 completion 배열만큼 반복문을 돌려주고, 완주자 completion 배열의 요소를 접근하여 기존의 map에서 존재한다면 vaule값을 -1을 해줍니다.

---

• ["marina", "josipa", "nikola", "vinko", "filipa"], ["josipa", "filipa", "marina", "nikola"]
  • {marina=0, filipa=0, nikola=0, josipa=0, vinko=1}
• ["mislav", "stanko", "mislav", "ana"], ["stanko", "ana", "mislav"]
  • {ana=0, mislav=1, stanko=0}

위와 같이 map이 형성되는 데 값이 1인 참가자가 완주를 하지 못한 참가자가 됩니다. 

 

마지막으로 향상된 for문과 keySet() 메서드를 사용하여 각 참가자의 키를 가져오고 get() 메서드와 함께 사용하여 값을 가져와 해당 값이 0이 아닌 즉 완주하지 못한 참가자를 sb에 append()하여 넣어줍니다.

 

 

refactoring ✅

import java.util.*;

class Solution {
    public String solution(String[] participant, String[] completion) {
        String answer = "";
        HashMap<String,Integer> map = new HashMap<>();
        StringBuilder sb = new StringBuilder();

        for(String name : participant){
            map.put(name, map.getOrDefault(name, 0) + 1);
        }

        for(String name : completion){
            map.put(name,map.get(name) - 1);
        }

        for(String key : map.keySet()){
            if(map.get(key) != 0){
                sb.append(key);
            }
        }

        return sb.toString();
    }
}

 

전체적인 코드의 맥락은 "나의 풀이"와 동일합니다. 그러나 NULL체크를 하는 부분이 없는 것을 볼 수 있습니다.

 

이는 HashMap의 getOrDefault() 메서드가 제공하는 "기본값" 때문입니다.

 

getOrDefault 메서드를 사용하면 특정 키에 대한 값이 존재하지 않을 때 지정한 기본값을 반환합니다. 따라서 null 체크를 할 필요 없이 키가 존재하지 않을 경우의 값을 쉽게 처리할 수 있습니다.

 

 

참고 ❗

[JAVA] 입출력, BufferedReader, StringTokenizer, StringBuilder 알아보기

[JAVA] HashMap 이란 ❓ (1/2)