[프로그래머스 레벨1] 완주하지 못한 선수 - JAVA(Hash 사용)

문제

 

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

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

 

 

제한사항

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

 

입출력 예

participantcompletionreturn

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"

 

 

나는 위 문제를 이중 for 문으로 해결했다. 

return 값을 제대로 받았지만 효율성에서 탈락해서 해당 문제를 풀지 못했다.

 

일단 소스를 보자.

class Solution {
    public String solution(String[] participant, String[] completion) {
        String answer = "";
        
        for(int i = 0; i<participant.length;i++){
            boolean flag = true;
            for(int j = 0; j<completion.length;j++){
                //같은게 나오면 flag false 주고 break
                if(participant[i].equals(completion[j])){
                    flag = false;
                    completion[j] = "";
                    break;
                }
            }
            if(flag){ //true라면 바뀌지 않은거니 통과 못한애임
                answer += participant[i];
            }
        }
        return answer;
    }
}

 

효율성에서 떨어지는 이유는 시간 복잡도가 n*n이기 때문이다.

 

이를 해결 하기 위해선 일단 for문을 1개로 만들어야 하고 뭔가 다른 방식을 사용해야 한다.

혼자 이 문제를 해결하기로 했지만 처음 코딩테스트 를 준비해본 터라 일단은 답지를 보고 소스를 리뷰해보기로 했다.

 

다른 소스들은 보니 효율이 제일좋은게 hashmap을 사용하는거라고 했다. 

해당 소스는 아래와 같다.

import java.util.*;
class Solution {
    public String solution(String[] participant, String[] completion) {
        String answer = "";
        int val = 0;
        
        Map<String, Integer> hm = new HashMap<>();
        
        //참가자를 Map에 저장하는 부분
        for(String part : participant){
            if(hm.get(part) == null){ //참가자 이름으로 저장되어 있는게 없으면 진입
                //Map에 (이름, 1) 이렇게 저장
                hm.put(part,1);
            }else{ //참가자가 Map에 저장되어 있을때 진입(중복 이름이 발생했을때)
                //해당 이름으로 저장된 val 값에 +1을 해줌
                val = hm.get(part) + 1;
                //Map에 (이름, 기존val+1) 이렇게 저장
                hm.put(part,val);
            }
        }
        
        //완료한 사람의 val 값을 -1 해서 빼주는 부분
        for(String comp : completion){
            //완료한 사람의 이름으로 val값을 가져와서 -1 시켜줌 (완료한사람은 빼는 과정)
            val = hm.get(comp) - 1;
            //Map에 (이름, 기존val-1) 이렇게 저장
            hm.put(comp,val);
        }
        
        for(String key : hm.keySet()){ //keySet을 사용해서 key들만 뽑은 set으로 반환
            //key값 즉 이름으로 되어 있는 사람중 1이 남아 있는사람(완주하지 못한사람) 진입
            if(hm.get(key) == 1){
                //answer에 해당 참가자 이름(키값)을 집어 넣음
                answer = key;
            }
        }
        return answer;
    }
}

내가 아래 소스에 주석을 달았다.

 

결과는 통과!!

 

이걸 생각해내다니...대단한거같다.

나도 저런 날이 올때까지 계속해서 코딩하자.