[Java] 프로그래머스. 완주하지 못한 선수

문제 설명

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

마라톤에 참여한 선수들의 이름이 담긴 배열 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. LinkedList 사용
   사실 링크드 리스트는 처음 코드 자체가 왜 효율성 테스트를 통과하지 못하는지 의문이었는데 두 가지 문제가 있다는 걸 알았다. 첫번째는 remove()의 평균 시간복잡도가 O(n)이라는 것, 두번째는 리스트가 정렬이 되어있냐에 따라 속도 차이가 많이 난다는 것. 여기서 remove()를 사용하지 않고는 어떻게 짜야할 지 모르겠어서 정렬만 했는데 바로 통과했다.
2. HashMap 사용
   처음 코드에서 효율성 테스트만 실패했는데 불필요한 탐색 절차(remove할 때 key 값 찾기 등)가 너무 많은 것 같아서 수정을 했다. 먼저 hashmap key 값에는 참여자 명단을, value 값에는 해당 이름의 참가자 수를 넣어준다. 그리고 완주자를 hashmap에서 remove하는 방식이 아니라 value값을 1 줄이며 덮어씌워준다.

정답(실패; 효율성 테스트)

import java.util.*;
class Solution {
    public String solution(String[] participant, String[] completion) {
        HashMap<String, Integer> hashmap = new HashMap<>();
        for (String s : participant) {
            hashmap.put(s, hashmap.getOrDefault(s, 0) + 1);
        }

        for (String s : completion) {
            for (Map.Entry<String, Integer> entry : hashmap.entrySet()) {
                if (entry.getKey().equals(s)) {
                    entry.setValue(entry.getValue() - 1);
                    break;
                }
            }
        }

        String[] result = hashmap.entrySet().stream()
                .filter(i -> i.getValue() == 1)
                .map(Map.Entry::getKey)
                .toArray(String[]::new);
        
        return result[0];
    }
}

정답(실패; 효율성 테스트)

import java.util.*;
class Solution {
    public String solution(String[] participant, String[] completion) {
        LinkedList<String> list = new LinkedList<>();
        for(String s : participant) {
            list.add(s);
        }

        for(String s : completion) {
            list.remove(String.valueOf(s));
        }

        return list.get(0);
    }
}

정답(LinkedList 사용)

import java.util.*;
class Solution {
    public String solution(String[] participant, String[] completion) {
        Arrays.sort(participant);
        Arrays.sort(completion);
        
        LinkedList<String> list = new LinkedList<>();
        for(String s : participant) {
            list.add(s);
        }

        for(String s : completion) {
            list.remove(String.valueOf(s));
        }

        return list.get(0);
    }
}

정답(HashMap 사용)

import java.util.*;
class Solution {
    public String solution(String[] participant, String[] completion) {
        String result = "";
        HashMap<String, Integer> hashmap = new HashMap<>();
        
        for (String s : participant) {
            hashmap.put(s, hashmap.getOrDefault(s, 0) + 1);
        }

        for (String s : completion) {
            hashmap.put(s, hashmap.get(s)-1);
        }

        for(String key : hashmap.keySet()) {
            if(hashmap.get(key)!=0) result = key;
        }
        
        return result;
    }
}