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

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

프로그래머스

 

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

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

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

입출력 예

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

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

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

---

import java.util.*;
class Solution {
    public String solution(String[] participant, String[] completion) {
        String answer = "";
        
        HashMap<String, Integer> map = new HashMap();
        //참가자 명단을 맵에 추가한다. 키는 이름, 값은 인원 수
        for(String person : participant){
            map.put(person, map.getOrDefault(person, 0) + 1);
        }
        //완주자명단을 맵에서 제외한다.
        for(String person : completion){
            map.put(person, map.get(person) - 1);
        }
        //남은 인원 중 완주하지 못한 선수를 찾아서 반환한다.
        for(String person : map.keySet()){
            if(map.get(person) > 0){
                answer = person;
                break;
            }
        }
        return answer;
    }
}

 

이중반복문을 사용해서 문제를 풀 수 있지만 시간복잡도때문에 효율적인 풀이가 될 수 없다.

그럴 땐 해시맵 사용!!

 

map의 key에는 participant배열의 선수 이름을 넣어주고, value에는 getOrDefault 함수를 사용해서 넣어준다.

getOrDefault() 사용해서 키가 존재하면 값에 1을 더하고, 존재하지 않으면 0을 반환한 후 1을 더한다.

 

다시 반복문을 사용해서 완주자 명단을 확인하고 map에서 선수의 인원수를 하나씩 뺀다.

 

마지막으로, map에 남은 선수들 중 인원수가 1 이상인 경우를 찾는다. 

완주하지 못한 선수의 이름은 인원수가 1이니 answer에 선수 이름을 넣어주고 중지시킨다!