수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.
마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.
- 마라톤 경기에 참여한 선수의 수는 1명 이상 100,000명 이하입니다.
- completion의 길이는 participant의 길이보다 1 작습니다.
- 참가자의 이름은 1개 이상 20개 이하의 알파벳 소문자로 이루어져 있습니다.
- 참가자 중에는 동명이인이 있을 수 있습니다.
이 문제는 해시 알고리즘을 이용한 문제이기 때문에 딕셔너리를 이용해서 문제를 풀었다. 매우 쉬운 난이도지만 푸는 방식이 다양해서 글을 쓰게 되었다. 내가 처음 구현한 방식은 아래와 같다.
- participant의 선수들을 key로 하는 딕셔너리 생성
- 각 선수들의 이름 수를 세어 value로 저장
- completion의 선수들의 수를 빼서 0이 되는 값 리턴
def solution(participant, completion):
dic = {}
for p in participant:
dic[p] = dic.get(p, 0) + 1
for c in completion:
dic[c] -= 1
dnf = [k for k in dic if dic[k] != 0]
answer = dnf[0]
return answer
이 방식 이외에 다른 사람이 풀었던 코드 중 기억에 남는 코드가 있었는데 바로 파이썬 내장 함수 hash를 이용해서 푸는 경우이다.
- participant의 선수의 hash 값을 key로, 선수 이름을 value로 하여 딕셔너리에 저장
- hash 값을 전부 더해줌
- 전체 더한 hash 값에서 completion의 hash 값을 빼줌
- 마지막 남은 hash 값의 value가 완주하지 못한 선수
def solution(participant, completion):
hashDict = {}
sumHash = 0
for i in participant:
hashDict[hash(i)] = i
sumHash += hash(i)
for j in completion:
sumHash -= hash(j)
return hashDict[sumHash]
내가 푼 코드만 보는 것이 아니라 다른 사람이 어떻게 풀었는지 보는 것도 코딩테스트를 준비할 때 매우 중요한 것 같다. 다른 사람이 푼 방식도 기억해두고 다시 한번 풀어보는 연습도 해봐야겠다.