[Algorithm][Programmers] 완주하지 못한 선수 (Java)

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/42576

프로그래머스

문제 설명

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

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

입출력 예시

풀이

HashMap을 사용하여 Key는 선수들 이름, Value는 인원 수로 설정하여 완주한 선수가 있으면 Value - 1 해주고 최종적으로 KeySet()을 활용하여 Map 전체를 돌면서 Value값이 0이 아니면 완주하지 못한 선수이므로 answer에 값을 넣고 break로 설정해주었습니다.

동일한 이름이 있을 수 있기 때문에 getOrDefault()함수를 사용하여 기존 키 값에 +1을 해주었습니다.

Map을 전체 출력하는 다양한 방법들이 있는데 제일 간편한 방법으로는 KeySet()을 사용하지만 Key를 이용하여 Value를 찾는 과정에서 시간이 소요될 수 있기 때문에 데이터가 많을 경우 entrySet()을 사용하는 것이 더 효율적입니다.

// 1. KeySet() 사용
for(String name : members.keySet()) {
    if(members.get(name) != 0) {
        answer = name;        
        break;
    }
}
// 2. entrySet() 사용
for(Map.Entry<String, Integer> entry : members.entrySet()) {
    if(entry.getValue() != 0) {
        answer = entry.getKey();
        break;
    }
}
// 3. Iterator 사용
Iterator<Map.Entry<String, Integer>> entries = members.entrySet().iterator();
while(entries.hasNext()) {
    Map.Entry<String, Integer> entry = entries.next();
        if(entry.getValue() != 0) {
        answer = entry.getKey();
        break;
    }
}

코드

import java.util.*;
class Solution {
    public String solution(String[] participant, String[] completion) {
        HashMap<String, Integer> members = new HashMap<>();
        String answer = "";
        for(int Index = 0; Index < participant.length; Index++) {
            members.put(participant[Index], members.getOrDefault(participant[Index], 0) + 1);
        }    
        for(int Index = 0; Index < completion.length; Index++) {
            members.put(completion[Index], members.get(completion[Index]) - 1);
        }
        for(String name : members.keySet()) {
            if(members.get(name) != 0) {
                answer = name;        
                break;
            }
        }
        
        return answer;
    }
}

제한사항