수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.
마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.
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" |
import java.util.*;
class Solution {
public String solution(String[] participant, String[] completion) {
String answer = "";
HashMap<String, Integer> h = new HashMap<>();
for (String player : participant) h.put(player, h.getOrDefault(player, 0) + 1);
for (String player : completion) h.put(player, h.get(player) - 1);
for (String key : h.keySet()) {
if (h.get(key) != 0){
answer = key;
}
}
return answer;
}
}
우선 HashMap h를 선언해주고 그 안에 participant에 있는 선수들과 1 값을 넣어준다. 그리고 다음 for
문에서 completion에 선수가 존재한다면 다시 1을 빼준다. 그렇게 하면 완주한 선수의 값은 0이 되고 완주하지 못한 선수는 1값을 가지고 있게 된다. 따라서 값이 0이 아닌 선수를 answer에 담아 리턴해주면 된다.