[프로그래머스] 완주하지 못한 선수(LV.1) - 파이썬(Python)

프로그래머스 > 완주하지 못한 선수

문제 설명

수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.
마라톤에 참여한 선수들의 이름이 담긴 배열 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"

입출력 예 설명

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

---

👊 내 문제 풀이

먼저 sort() 함수를 사용해 두 배열 정렬한다. 완주자의 수만큼 차례대로 두 배열을 비교하고, 값이 같지 않은 경우 해당 인덱스의 참가자를 return 한다. 위 과정에서 두 배열의 모든 인덱스가 같다면, 마지막 참가자가 완주하지 않았다는 뜻이므로 참가자 배열의 마지막 인덱스를 return 하면 된다.

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

채점 결과

채점 결과는 통과지만 과연 이렇게 풀라고 낸 문제일까? 하는 의구심이 들어 풀이를 확인했다.

 

💡 다른 사람의 풀이

1. collections 모듈 사용

collections.Counter는 컨테이너에 들어있는 요소의 개수를 {'이름' : '개수'} 와 같은 딕셔너리 형태로 반환해준다.

다음은 collections.Counter의 활용 예시다.

import collections

participant = ['amy', 'jane', 'mike', 'nicole']
completion = ['amy', 'jane', 'nicole']

>>> print(collections.Counter(participant))
Counter({'jane': 1, 'amy': 1, 'nicole': 1, 'mike': 1})

>>> print(collections.Counter(completion))
Counter({'jane': 1, 'amy': 1, 'nicole': 1})

>>> print(collections.Counter(participant) - collections.Counter(completion))
Counter({'mike': 1})

Counter 객체는 뺄셈을 지원한다는 특이점을 활용하여 문제를 푼다.

완주를 못한 인원은 1명이기 때문에, 완주자에서 참가자를 빼서 얻은 딕셔너리의 0번째 key 값을 return 하면 된다.

import collections

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

 

2. zip() 내장함수 사용

접근은 내 풀이와 동일하나, zip함수를 사용해 두 배열을 묶어 더욱 간단하게 비교한다.
zip()은 2개 이상의 리스트들을 동일한 index끼리 묶어주는 함수다.
이 때, 리스트의 길이가 동일하지 않을 경우 짝이 완성되지 않는 인덱스는 버린다.

다음은 zip() 함수의 활용 예시다.

a = ['a', 'b', 'c', 'd']
b = ['apple', 'banana', 'carrot']

list(zip(a, b))
>>> [('a', 'apple'), ('b', 'banana'), ('c', 'carrot')]

def solution(participant, completion):
    participant.sort()
    completion.sort()
    for p, c in zip(participant, completion):
        if p != c:
            return p
    return participant[-1]

sort() 함수
len() 함수
zip() 함수
collections 모듈