프로그래머스 | 완주하지 못한 선수 [파이썬 python]

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

처음 시도한 코드

def solution(participant, completion):
    for i in completion:
        if i in participant:
            participant.remove(i)
    return ''.join(participant)

어렵지 않은 문제이기 때문에 쉽게 코드를 작성할 수 있었다.

하지만 정확성 테스트는 모두 통과하지만 효율성 테스트는 시간 초과로 실패하였다.

내가 작성한 코드에서 시간을 많이 소요되는 부분을 생각해 보니 for문과 remove임을 알 수 있었다.

completion이 길어지면 길어질수록 for문의 시간도 증가하고, participant가 길어질수록 remove에도 시간이 오래 걸린다.

정답 코드

from collections import Counter

def solution(participant, completion):
    p = Counter(participant)
    c = Counter(completion)
    fail = list(p-c)
    return fail[0]

그래서 요소 하나를 두 리스트에 공통으로 존재하는지 비교하는 것보단 Counter를 사용하는 방식을 택하였다.

Counter를 사용하여 participant, completion의 모든 요소들의 개수를 센 뒤, 차를 구하면 해당 값이 완주하지 못한 선수가 된다.

Counter끼리 마이너스(-) 연산이 가능하기 때문에 쉽게 완주하지 못한 선수를 찾을 수 있다.