완주하지 못한 선수

완주하지 못한 선수

문제풀러가기

단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.

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

제한사항

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

def solution(participant, completion):
    answer = ''
    comp = completion
    for part in participant:
        if part not in comp:
            answer+= part
        else:
            comp.remove(part)
    return answer

이라고 작성한 코드는 효율성 테스트를 통과하지 못했다.

list 아이템을 찾아 삭제하는 remove 함수가 O(n)시간이 걸려서 총 시간복잡도가 O(n^2)되는 듯 하다.

def solution(participant, completion):
    p = sorted(participant)
    c = sorted(completion)
    # print(len(p), len(c))
    for i in range(len(p)-1):
        if p[i] != c[i]:
            return p[i]
    return p[-1]

파이썬의 함수별 시간복잡도 도큐멘테이션에 따르면 정렬함수는 O(nlogn)시간이 걸린다.

매 iteration마다 삭제연산을 해 총 시간복잡도 O(n^2)이 나오느니, 총 시간복잡도 O(nlogn)+O(nlogn)이 났겠다싶어서, 두 배열을 sorted 함수로 정렬 한 뒤 O(n)시간 만큼 단순 비교연산을 수행했다.

정확성 테스트

테스트 1 〉	통과 (0.05ms, 12.2MB)
테스트 2 〉	통과 (0.05ms, 12.1MB)
테스트 3 〉	통과 (0.33ms, 12.2MB)
테스트 4 〉	통과 (0.70ms, 12.2MB)
테스트 5 〉	통과 (0.69ms, 12.3MB)

효율성 테스트

테스트 1 〉	통과 (53.58ms, 89.2MB)
테스트 2 〉	통과 (87.81ms, 127MB)
테스트 3 〉	통과 (124.47ms, 155MB)
테스트 4 〉	통과 (119.66ms, 170MB)
테스트 5 〉	통과 (134.45ms, 170MB)

로 효율성 테스트는 통과했지만, collections.Counter 함수를 사용한 쪽이 더 간결해보이고, 효율성도 조금 더 좋다.

import collections


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

정확성 테스트

테스트 1 〉	통과 (0.15ms, 12MB)
테스트 2 〉	통과 (0.13ms, 12.1MB)
테스트 3 〉	통과 (0.36ms, 12.2MB)
테스트 4 〉	통과 (0.60ms, 12.3MB)
테스트 5 〉	통과 (0.60ms, 12.4MB)

효율성 테스트

테스트 1 〉	통과 (34.02ms, 89.2MB)
테스트 2 〉	통과 (43.79ms, 129MB)
테스트 3 〉	통과 (78.56ms, 154MB)
테스트 4 〉	통과 (68.90ms, 170MB)
테스트 5 〉	통과 (86.31ms, 169MB)