Python/알고리즘

프로그래머스-파이썬 (완주하지 못한 선수)

SeongWon 2021. 1. 12. 19:58
반응형

Q. 수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다. 마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.

 

 

<제한사항>

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

 

<예시>

 

 

 

 

<해결과정>

 

처음에는 단순히 리스트를 다루는 문제라고 생각했다. 그러면 매우 단순한 문제라고 생각이 들 수 있다.

나도 그랬기에 for문을 이용하여 completion 리스트의 요소 중, participant 요소에 존재하지 않는 것을 반환하였다.

이는 정확성은 통과하지만, 요소 수가 수 만개가 된다면 이 로직은 성능면에서 매우 떨어져서 효율성 측면이 0점 이였다.

 

그래서 for문 없이 해결할 방법이 없을까 하다가 찾아본 방법이 'Counter' 라는 함수를 이용하는 것이였다.

Counter는 collection 모듈에 내장되어 있는 클래스로 특정 요소와 그 요소의 개수를 카운팅하여 '요소 : 개수' 의 

딕셔너리 형태로 반환해준다.

 

예시)

 

from collections import Counter

completion = ["josipa", "filipa", "marina", "nikola" , "nikola"]

print (Counter(completion))
#결과: Counter({'josipa': 1, 'filipa': 1, 'marina': 1, 'nikola': 2})

 

 

<내 풀이>

from collections import Counter

def solution(participant, completion):
    count_p = Counter(participant)
    count_c = Counter(completion)
    
    count_p.subtract(count_c)
    
    answer = count_p.most_common(1)[0][0]
    
    return answer

 

※ 제한사항을 보면 완주하지 못한 선수는 참가자 중, 반드시 1명만 나오게 된다.

   즉, 이름이 중복되있던, 안돼있던 참가자의 명단에서 완주자의 명단을 빼면 1명이 남게된다.

 

subtract() : 요소의 차를 구한다. 없는 요소를 빼게 될 경우, 음수로 나오게 된다.

most_common() : 등장하는 요소의 빈도 수가 많은 순으로 정렬하여 튜플로 반환해준다.

 

참가자 명단에서 완주자 명단을 빼는 과정을 거치면 등장 빈도 수가 1인 요소가 하나만 남게되고, 이를 most_common() 함수로 정렬하면 가장 첫 번째로 등장하는 선수가 바로 완주하지 못한 선수가 되는 것이다.

반응형