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

코딩테스트 연습 - 완주하지 못한 선수

import sys

input = sys.stdin.readline
participant = list(map(str, input().split()))
completion = list(map(str, input().split()))


def solution(participant, completion):
    participant.sort()
    completion.sort()
    for i in range(len(completion)):
        if completion[i] != participant[i]:
            return participant[i]
    return participant[-1]


print(solution(participant, completion))

 

해설

처음엔 단순하게 completion에는 있지만 participant에 없는 원소를 출력했는데 동명이인이라는 경우를 생각하지 못해 틀렸다.

두 리스트를 정렬 후, 같은 인덱스에 서로 일치하지 않는 데이터를 리턴해줬다. completion의 모든 데이터를 탐색해도 없을 경우는 participant의 마지막 값이 답이므로 이 경우에는 participant의 마지막 값을 리턴했다.