[Programmers/Python] 42576_완주하지 못한 선수

프로그래머스

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

문제 설명 )

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

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

제한사항 )

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

입출력 예 )

입출력 예 설명 )

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

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

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

 

📌 문제 풀이

단순히 반복문을 통해서 participant의 선수가 completion에도 존재하는 지를 체크하게 된다면,
동일한 이름의 참여자가 여러 명일 경우에서 문제가 발생한다.
따라서, 각각 해당 선수의 이름의 개수를 비교하여 일치하지 않는다면, 일치하지 않는 선수의 이름을 반환하면 된다.
 
먼저, 시간 초과로 실패한 코드들부터 살펴보자면,, 😂😂

def solution(participant, completion):
    for i in participant:
        if i not in completion:
            return i
        
        else:
            completion.pop(completion.index(i))

- 완주자 목록에 존재하지 않는 이름은 그대로 바로 반환하고, 그렇지 않다면, 해당 선수를 pop하여 삭제처리 해주었다.
- for문이 돌 때마다 completion에서 O(N)의 시간복잡도가 발생하여 결국 $O(N^2)$의 시간 복잡도를 가지게 된다.
따라서 시간 초과로 실패!
 

def solution(participant, completion):
    for i in participant:
        if participant.count(i) != completion.count(i):
            return i

참여자 목록에서의 해당 이름의 개수와 완주자 목록에서의 개수를 비교하여 일치하지 않는다면 반환이 되도록 하였다.
문제는 이 코드도 마찬가지로 count로 개수를 세어주기 위해 $O(N^2)$의 시간복잡도가 반복된다는 점이다.
따라서 시간 초과로 실패!

---

count와 collections 모듈의 Counter의 시간 복잡도를 비교해보면, 
count는 각 원소의 개수를 세어주기 위해 $O(N^2)$ 이고, Counter는 O(N)이다.
따라서 시간 초과를 Counter를 통해 해결해줄 수 있었다.

✅ [ CODE ]

# solution_1
from collections import Counter

def solution(participant, completion):
    participantCnt = Counter(participant)
    completionCnt = Counter(completion)

    for i in participant:
        if participantCnt[i] != completionCnt[i]:
            return i

Counter로 각 선수의 이름의 개수를 비교한 후, 다르다면 반환해준다.

---

이 문제의 카테고리는 해시 함수였기에, key와 value를 통해서 답을 구하는 솔루션은 다음과 같이 구현해보았다.
1. key 값으로 설정된 적이 없는 선수의 이름을 1과 함께 저장한다.
2. 설정된 적 있는 선수의 이름이라면, +1로 증가시켜준다.
3. 완주자 목록에 따라서, 참여자 목록에 있던 이름이라면 -1을 해준다.
4. 0이 되어 확인이 모두 완료된 경우, 저장된 data에서 삭제처리 해준다.
5. 따라서, 완주자 목록에 존재하지 않는 이름은 그대로 반환해준다.

 

✅ [ CODE ]

# solution_2
def solution(participant, completion):
    data = dict()
    for i in participant:
        if data.get(i, 0) == 0:
            data[i] = 1
        else:
            data[i] += 1
    
    for i in completion:
        if data.get(i, 0) == 0:
            return i
        else:
            data[i] -= 1
            if data[i] == 0:
                data.pop(i)
    return list(data.keys())[0]