36일차. 완주하지 못한 선수

링크

https://school.programmers.co.kr/learn/courses/30/lessons/42576

프로그래머스

 

문제

문제 설명

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

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

제한사항

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

 

정답

from collections import Counter

def solution(participant, completion):
    try:
        return list(set(participant)-set(completion))[0]
    except:
        cp = Counter(participant)
        cc = Counter(completion)
        for s in set(completion):
            if cp[s] != cc[s]:
                return s

 

풀이과정

 

일단 중복이 없는경우엔 set으로 뺴버리면 아주 쉽게 끝난다.

return list(set(participant)-set(completion))[0]

문제는 중복이 있는경우. 이 경우에는 set이 중복을 제거해버리므로, 빈 리스트를 인덱싱해서 out of index 에러가 난다.
이를 추출하기위해 try로 묶어주었다.

이번에는 collections의 Counter를 사용할것이다.

from collections import Counter
cp = Counter(participant)
cc = Counter(completion)
for s in set(completion):
    if cp[s] != cc[s]:
        return s

중복이 없는 각 요소마다 갯수를 비교해서 다르면 return해버리는방법이다. 둘의 요소가 완전히 동일하지 않을 경우는 위에서 제거했기떄문에 없다.

처음부터 counter로 한다음 -로 계산햇어도 결과는 나온다. 하지만 set이 기본적으로는 더 효율적이기 떄문에 효율성 검사를 고려해서 이런 방식으로 짰다.