완주하지 못한 선수

https://programmers.co.kr/learn/courses/30/lessons/42576

코딩테스트 연습 - 완주하지 못한 선수 | 프로그래머스

카테고리: 해시

 

수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다. 마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요. 마라톤 경기에 참여한 선수의 수는 1명 이상 100,000명 이하입니다. completion의 길이는 participant의 길이보다 1 작습니다. 참가자의 이름은 1개 이상 20개 이하의 알파벳 소문자로 이루어져 있습니다. 참가자 중에는 동명이인이 있을 수 있습니다.

 

카테고리가 hash라는 것이 가장 큰 힌트가 되었던 것 같다.

중복되는 값이 있을 수 있는 경우 hash를 사용하는 것이 시간 효율을 높일 수 있다.

 

어차피 hash에 값을 다 넣어야되니 linear search가 일어나는 것은 필연적이다.

처음에는 participant를 hash map에 추가하고, completion을 검색하며 hash에서 지우려고 했는데,

생각해보니 completion을 넣어두고 participant를 검색하던 중 발견되지 않으면 바로 정답이 도출된다.

따라서 우선 completion의 모든 값을 hash map에 추가하고,

발견된 key들은 값을 줄이며 값이 0이 되는 경우 hash에서 삭제하여 검색 속도를 더욱 높이는 방식으로 구현했다.

 

import java.util.HashMap;

class Solution {
    public String solution(String[] participant, String[] completion) {
        var completionList = new HashMap<String, Integer>();
        
        // put all completion in completionList
        for (String c : completion) {
            if (completionList.containsKey(c)) {
                int originalValue = completionList.get(c);
                completionList.put(c, originalValue + 1);
            }
            else {
                completionList.put(c, 1);
            }
        }
        
        for (String p : participant) {
            if (completionList.containsKey(p)) {
                int originalValue = completionList.get(p);
                completionList.put(p, originalValue - 1);
                
                if (originalValue < 2) {
                    completionList.remove(p);
                }
            }
            else {
                return p;
            }
        }
        return null;
    }
}