[프로그래머스/파이썬] (해시) 완주하지 못한 선수

출처

문제 설명

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

마라톤에 참여한 선수들의 이름이 담긴 배열 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"

입출력 예 설명

예제 #1

"leo"는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.

예제 #2

"vinko"는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.

예제 #3

"mislav"는 참여자 명단에는 두 명이 있지만, 완주자 명단에는 한 명밖에 없기 때문에 한명은 완주하지 못했습니다.

내 풀이

def solution(participant, completion):
    answer = ''
    p = participant
    c = completion
    p.sort()
    c.sort()
    for i in range(len(p)):
        if i == len(p)-1:
            answer = p[i]
            break
        if p[i]!=c[i]:
            answer = p[i]
            break
    return answer

p와 c를 정렬했기 때문에 완주하지 못하는 참가자가 생기면 p[i]와 c[i]가 달라질 것이라 생각했다.
for문의 두번째 if문이 이를 표현했다.
만약에 p의 마지막 선수가 완주를 못했다면 for문이 돌아 i가 len(p)-1까지 갈 것이다.
따라서 i가 len(p)-1이면 마지막 선수가 완주를 못했다는 것을 표현한 게 첫번째 if문이다.

효율성 테스트: ~96.65ms / 26.4MB

다른 풀이

import collections

def solution(participant, completion):
    answer = collections.Counter(participant) - collections.Counter(completion)
    return list(answer.keys())[0]

collections 모듈의 Counter 함수를 사용했다. 코린이인 나는 물론 처음 보지만 앞으로 자주 쓰일 것 같다.

카운터 함수

import collections

print(collections.Counter(["leo", "kiki", "eden", "eden"])) 
# Counter({'eden': 2, 'leo': 1, 'kiki': 1})

Counter 함수는 리스트의 각 요소가 몇 번 등장했는지 말 그대로 카운트 해주는 함수인 듯하다.
위 풀이에서 participant의 Counter와 completion의 Counter를 빼면 Counter({'완주하지 못한 선수': 1}) 가 출력될 것이다.

solution함수의 return 값으로 ['완주하지 못한 선수']를 반환해야 하므로 Counter의 내장함수 keys를 이용하여 list에 담는다.