[Programmers] 완주하지 못한 선수 (python)

문제

https://school.programmers.co.kr/learn/courses/30/lessons/42576

프로그래머스

문제 설명

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

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

제한사항

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

입출력 예

["leo", "kiki", "eden"]	["eden", "kiki"]	"leo"
["marina", "josipa", "nikola", "vinko", "filipa"]	["josipa", "filipa", "marina", "nikola"]	"vinko"
["mislav", "stanko", "mislav", "ana"]	["stanko", "ana", "mislav"]	"mislav"

 

풀이과정

풀이 1:  dict 자료구조를 이용한 풀이 → 효율성 통과하지 못함

1. 이름을 key, 리스트 내의 횟수를 value로 하여 participant, completion 각각 dict 생성 → p_dict, c_dict
2. p_dict, c_dict의 key들을 정렬하여 같은지 비교
3. 같으면 각 키의 value가 같지 않은 key를 반환
4. 같지 않으면 c_dict의 key에 포함되지 않는 p_dict의 key를 반환

풀이 2:  자료구조 변환 없이 주어진 리스트를 정렬, 하나씩 비교하여 값이 다른 순간 반환 → 효율성 통과

1. participant, completion을 정렬
2. answer을 빈 문자로 초기화
3. completion과 participant의 요소를 하나씩 비교
4. 다른 값이 경우의 participant 요소를 반환
5. completion의 마지막까지 확인했을 때도 answer이 빈 문자열이면 participant의 마지막 값을 반환

 

Script

풀이 1

def solution(participant, completion):
    p_dict = {k: participant.count(k) for k in participant}
    c_dict = {k: completion.count(k) for k in completion}

    if list(sorted(p_dict.keys())) == list(sorted(c_dict.keys())):
        for k, v in c_dict.items():
            if p_dict[k] != v:
                answer = k
                return answer
    else:
        answer = [x for x in p_dict.keys() if x not in c_dict.keys()][0]


    return answer

 

풀이 2 : 실제 제출

def solution(participant, completion):
    p_sorted = sorted(participant)
    c_sorted = sorted(completion)

    answer = ''

    for i, c_elem in enumerate(c_sorted):
        if p_sorted[i] != c_elem:
            print(p_sorted[i])
            answer = p_sorted[i]
            return answer
    if answer == '':
        answer = p_sorted[-1]
        return answer