[프로그래머스 / Python] 완주하지 못한 선수(해시) - Lv.1

완주하지 못한 선수(해시) - Lv.1

수많은 마라톤 선수들이 마라톤에 참여하였습니다.

단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.

 

마라톤에 참여한 선수들의 이름이 담긴 배열 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):
    for i in completion:
        if i in participant:
            participant.remove(i)
    return participant[0]

처음엔 set으로 단순히 participant에서 completion을 빼는걸로 해결하고자 했는데 문제 조건에 동명이인이 존재하여 set을 쓰면 동명이인까지 없어져서 문제를 해결할 수 없었다.

 

그래서 완주자 목록을 for문으로 돌면서 참여자 목록에 있으면 참여자 목록에서 하나씩 제거하는 방식으로 해결하려 했으나, 효율성 테스트에서 통과하지 못했다. for문의 시간 복잡도가 O(N)이기 때문에 for문만 1번 사용하면 될거라고 생각했는데 그게 아니였다..

 

문제에서 참가자의 범위가 1명 이상 100,000명 이하인데 for문 뿐만 아니라 특정 값이 있는지 확인하는 in과 특정 값을 제거하는 remove의 시간 복잡도 역시 O(N)이기 때문에 통과하지 못하는 것 같다. 

 

결국 다른 방법을 생각했고 collections 모듈의 Counter 클래스를 사용하기로 했다.

from collections import Counter


def solution(participant, completion):
    answer = Counter(participant) - Counter(completion)

    return list(answer)[0]

이러면 동명이인이 존재해도 결과값에는 하나만 남게 되고, 속도도 훨씬 빨라졌다.

 

이전에 알고리즘 문제를 풀면서 이 문제와 같이 중복값을 세는 문제가 그 때 counter을 처음 알았는데, 앞으로도 유용하게 사용하게 될 것 같다!

---

다른 사람의 풀이

def solution(participant, completion):
    answer = ''
    temp = 0
    dic = {}
    for part in participant:
        dic[hash(part)] = part
        temp += int(hash(part))
    for com in completion:
        temp -= hash(com)
    answer = dic[temp]

    return answer

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

해시 함수를 사용하는 풀이는 신기하다.

 

아래 풀이는 내가 처음에 하려던 방식과 비슷한 것 같다. 두 리스트를 정렬한 뒤에 하나씩 비교하면서 다른 값이 있다면 return하고, completion을 다 돌아도 모든 값이 같다면 participant에 마지막으로 남은 한 가지 값이 정답이기 때문에 그 값을 return한다.

 

그래도 sort를 사용하기 때문에 내가 작성한 코드가 효율적이다. O(NlogN) < O(N)