[Hash / 해시] - 완주하지 못한 선수(HashMap)

문제

수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.

마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.

제한사항

• 마라톤 경기에 참여한 선수의 수는 1명 이상 100,000명 이하입니다.
• completion의 길이는 participant의 길이보다 1 작습니다.
• 참가자의 이름은 1개 이상 20개 이하의 알파벳 소문자로 이루어져 있습니다.
• 참가자 중에는 동명이인이 있을 수 있습니다.

입출력 예

participant	completion	return
["leo", "kiki", "eden"]	["eden", "kiki"]	"leo"
["marina", "josipa", "nikola", "vinko", "filipa"]	["josipa", "filipa", "marina", "nikola"]	"vinko"
["mislav", "stanko", "mislav", "ana"]	["stanko", "ana", "mislav"]	"mislav"

 

예제 #1
"leo"는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.

예제 #2
"vinko"는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.

예제 #3
"mislav"는 참여자 명단에는 두 명이 있지만, 완주자 명단에는 한 명밖에 없기 때문에 한명은 완주하지 못했습니다.

 

코드 1

import java.util.Map;
import java.util.HashMap;
class Solution {
    public String solution(String[] participant, String[] completion) {
        String answer = "";
        // 지원자 - 완료자 = 미완려자
        Map<String,Integer> hm = new HashMap<>();
        for(int i=0; i<participant.length; i++) {
        	hm.put(participant[i],hm.getOrDefault(participant[i], 0)+1);
        }    
        for(int j=0; j<completion.length; j++) {
        	hm.put(completion[j],hm.get(completion[j])-1);
        }
    	for(Map.Entry<String,Integer> en : hm.entrySet()) {
    		if(en.getValue() == 1) {
    			answer = en.getKey();
    		}
        }     
        return answer;
    }
}

 

코드 2

import java.util.Map;
import java.util.HashMap;
import java.util.Arrays;
class Solution {
    public String solution(String[] participant, String[] completion) {
        String answer = "";
        // 참여자
        Map<String, Long> participantMap = Arrays.stream(participant)
                .collect(HashMap::new, (map, name) 
                        -> map.merge(name, 1L, Long::sum), HashMap::putAll);
        // 완료자
        Arrays.stream(completion)
                .forEach(name -> participantMap.merge(name, -1L, Long::sum));

        answer =  participantMap.entrySet().stream()
                .filter(entry -> entry.getValue() > 0)
                .findFirst()
                .map(Map.Entry::getKey)
                .orElse(null);
        return answer;
    }
}

 

코드 2 - Map 인터페이스 메소드

`merge` 메서드는 `Map` 인터페이스에 속한 메서드 중 하나로, 두 개의 인자를 가지며 맵에 새로운 값을 적용합니다. 해당 메서드는 두 가지 연산을 수행하는 데 사용됩니다

map.merge(key, value, remappingFunction);

 

여기서:

- `key`: 맵에서 값을 추가 또는 업데이트할 키입니다.

- `value`: 해당 키에 추가하거나 업데이트할 값입니다.

- `remappingFunction`: 주어진 키와 연관된 현재 값 및 제공된 값으로 새 값을 계산하는 함수입니다.

 

이것을 이용하여 `participantMap.merge(name, 1L, Long::sum)`에서는 다음과 같은 일이 일어납니다:

- `name` 키가 이미 존재하면, 해당 키의 값을 `1L`과 더합니다.

- `name` 키가 존재하지 않으면, 새로운 키를 만들고 값을 `1L`로 설정합니다.

 

이렇게 함으로써 참가자 배열의 이름을 세는 효과를 내게 됩니다. 마찬가지로 `participantMap.merge(name, -1L, Long::sum)`에서는 완주자 배열의 이름을 세는 효과를 내게 됩니다.

 

이렇게 `merge` 메서드를 사용함으로써, 코드가 간결해지고 중복된 키의 관리가 편리해집니다.

코드 2 - collect

Map<String, Long> participantMap = Arrays.stream(participant)
        .collect(HashMap::new, (map, name) -> map.merge(name, 1L, Long::sum), HashMap::putAll);

`.collect(Collectors.toMap(...))` 부분은 Stream API를 사용하여 요소들을 수집하고 Map으로 변환하는데 사용됩니다. 

여기서 사용된 `Collectors.toMap()` 대신에 `.collect(HashMap::new, ...)`를 사용한 이유는 중복된 키가 있을 때, `merge` 메서드를 사용하여 값을 업데이트하기 위함입니다. 

        Map<String, Long> participantMap = Arrays.stream(participant)
                .collect(Collectors.groupingBy(name->name, Collectors.counting()));

- `HashMap::new`: 새로운 `HashMap`을 생성합니다.

- `(map, name) -> map.merge(name, 1L, Long::sum)`: 각 요소를 처리할 때마다 호출되는 람다 표현식입니다. 현재 요소를 `map`에 추가하고, 중복된 키가 있을 경우 기존 값과 1L을 더합니다.

- `HashMap::putAll`: 모든 매핑을 최종적으로 하나의 `HashMap`으로 병합합니다.

 

즉, 각 참가자의 이름을 키로 하고, 참가자의 수를 값으로 하는 `HashMap`을 생성합니다. 중복된 이름이 있으면 값을 더해가면서 생성됩니다.