코린스토리

[Programmers] Level 1 . 완주하지 못한 선수 파이썬 본문

Programmers Test/Level 1

[Programmers] Level 1 . 완주하지 못한 선수 파이썬

정선웅 2019. 12. 28. 18:54

문제 설명

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

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

제한사항

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

입출력 예

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

 

 

해답

def solution(participant, completion):

     d={}

     for x in participant:

         d[x]=d.get(x,0)+1

     for x in completion:

         d[x] -= 1

     dnf=[k for k, v in d.items() if v>0]

     answer=dnf[0]

     return answer

 

(Programmers 해답 참고함)

 

사유과정

해시를 사용해서 문제를 해결한 경우이다. 문제의 포인트는 참가자와 완주자는 한명의 차이가 있으며

'동명이인'이 있기 때문에 예제 3과 같은 경우(mislav가 완주자 명단에 있지만 동명이인이 있어서 탈락자에 포함되는 경우)가 생긴다

 

우선 간단하게 생각해보면 이름이 알파벳으로 이루어져 있기 때문에 sort를 사용하여 참가자, 완주자를

알파벳순으로 소팅하여 문제를 해결할 수 도 있다. 하지만 그렇게 할 경우 시간복잡도가 nlogn이 되기 때문에

다른 방식(Dictionary)을 사용하였다.

 

여기서 쓰인 Dictionary 함수는 items과 get이다.

 

처음에 d라는 Dictionary에 participant의 요소들을 받아와 1로 저장을 하고

completion의 요소들을 받아오면서 1을 제거해준다. 그렇게되면 '1'의 value를 가진 key가 완주하지 못한 사람의

이름이 될 것이다!