[프로그래머스] Python Lv.1 완주하지 못한 선수

 문제 설명  

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

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

제한사항

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

 

나의 풀이

def solution(participant, completion):
    p = set(participant)
    c = set(completion)
    if len(p-c) == 0:
        for j in p:
            if participant.count(j) > completion.count(j) :
                return j
    else:
        for i in p-c:
            return i

 

테스트를 모두 통과했으나 효율성 테스트 마지막 5번을 시간초과로 통과하지 못해 실패한 코드이다.

 

 

 

다른사람 풀이 + 개선점

 

Counter : 문자열에서 각 문자가 몇번씩 알려주는 객체가 반환됨

from collections import Counter

 

위와 같이 정의한 뒤 사용하면 된다.

 

예시)

from collections import Counter
print(Counter("I am happy to learn Python"))

Counter({' ': 5, 'a': 3, 'h': 2, 'p': 2, 'y': 2, 't': 2, 'o': 2, 'n': 2, 'I': 1, 'm': 1, 'l': 1, 'e': 1, 'r': 1, 'P': 1})

 

Counter를 사용하여 작성한 코드

from collections import Counter

def solution(participant, completion):
    part = Counter(participant)
    comp = Counter(completion)

    part.subtract(comp)

    return ''.join(list(part.elements()))

 

counter 를 통해 각 리스트의 문자열이 몇 개 있는지를 포함한 딕셔너리 형태의 part와  comp를 만들고

substract를 통해 part에서 comp의 단어들 개수를 뺀 것을 part로 지정해준다.

예시) 

["mislav", "stanko", "mislav", "ana"],["stanko", "ana", "mislav"]

participant와 completion이 위와 같을때 substract를 하면 아래와 같이 나온다

Counter({'mislav': 1, 'stanko': 0, 'ana': 0})

 

그 뒤 part.element를 통해 0 이 아닌 값을 빼온다 위와 같은 경우에는 'mislav' 가 되겠다.

list를 통해 part.element()를 담아줘야 ['mislav'] 의 형식으로 나온다.

그냥 part.element()를 출력하면 주소가 나온다.

따라서 위와 같이 list에 담아준뒤 join을 통해 문자열로 다시 빼내주는 작업을 거친다.

 

또는 list(part.keys())[0] 을 통해 part의 key값은 무조건 하나이므로 그 하나를 빼내주면 된다.