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

문제

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

마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때,

완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.

제한사항

(1) 마라톤 경기에 참여한 선수의 수는 1명 이상 100,000명 이하입니다.

(2) completion의 길이는 participant의 길이보다 1 작습니다.

(3) 참가자의 이름은 1개 이상 20개 이하의 알파벳 소문자로 이루어져 있습니다.

(4) 참가자 중에는 동명이인이 있을 수 있습니다.

입출력 예

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. 이름을 Key값으로, 같은 이름을 가진 사람의 수를 Value로 하는 Map 자료구조를 이용한다.

2. participant 배열을 탐색하며 Map에 이름이 없으면 Value를 1로 해서 넣어주고, 이미 있다면 해당 Key의 Value에 1을 더해서 Map의 Value를 변경해준다.

3. completion 배열을 탐색하며 Map에서 Key(이름)의 Value가 1이면 Map에서 삭제하고 1보다 크면 Value를 Value-1 값으로 변경해준다.

4. Map에 남아있는 Key를 return 해준다.

코드

package Programmers;
 
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
 
public class Solution_완주하지못한선수 {
 
    public static String solution(String[] participant, String[] completion) {
        String answer = "";
        Map<String, Integer> remain = new HashMap<>();
        for(int i=0; i<participant.length; i++) {
            if(remain.containsKey(participant[i])) {
                remain.put(participant[i], remain.get(participant[i])+1);
            } else remain.put(participant[i], 1);
        }
        for(int i=0; i<completion.length; i++) {
            if(remain.get(completion[i])>1) remain.put(completion[i], remain.get(completion[i])-1);
            else remain.remove(completion[i]);
        }
        Iterator<String> it = remain.keySet().iterator();
        while(it.hasNext()) answer = it.next();
        return answer;
    }
    
    public static void main(String[] args) {
        String[] participant = {"leo", "kiki", "eden"};
        String[] completion = {"kiki", "eden"};
        System.out.println(solution(participant, completion));
    }
}
 

코드에 대한 피드백이나 더 좋은 아이디어는 언제나 환영입니다.