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

❓문제

https://school.programmers.co.kr/learn/courses/30/lessons/42576?language=python3

프로그래머스

문제 설명

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

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

제한사항

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

✍🏻풀이

파이썬의 경우 딕셔너리를 이용하여 풀 수 있다.

먼저 동명이인이 있다고 하였기 때문에 딕셔너리를 이용해 각 참여자 이름별로 몇명인지 센다. 그리고 완주한 사람 중에 1 이상이면 1 빼고 1명만 있다면 아예 딕셔너리에서 제거한다.

그리고 문제에서 보면 단 한 명의 선수를 제외하고는 완주한다고 했으므로 딕셔너리에는 무조건 한명만 남는다. 그러므로 딕셔너리의  key만 가져오는데 이 때 딕셔너리 형태로 가져오기 때문에 list로 변형해서 인덱스 위치가 0인 것을 반환하면 된다.

💻코드

내 풀이

def solution(participant, completion):    
    dic = {}
    for p in participant:
        if p not in dic:
            dic[p] = 1
        else:
            dic[p] += 1
    
    for c in completion:
        if dic[c] > 1:
            dic[c] -= 1
        else:
            dic.pop(c)
    
    return list(dic.keys())[0]

다른 사람 풀이

import collections
# from collections import Counter

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

 

📍후기

다른 사람들은 collections에 있는 Counter를 이용해서 풀었다.

Counter에 대해 찾아보니 Counter는 중복된 데이터가 저장된 배열을 인자로 받아 각 원소가 몇 번씩 나오는지가 저장된 객체를 반환한다. 그리고 파이썬의 기본 자료구조인 딕셔너리를 확장하고 있어서 사전에서 제공하는 API를 그대로 사용할 수 있다고 한다.

그런데 Counter를 사용했을 때가 시간이 더 걸리고 메모리도 많이 차지한다. 물론 엄청나게 큰 차이는 아니지만...