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

2022. 2. 3. 00:55Computer Science/Algorithm

반응형

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

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

 

제한사항

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

입출력 예

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

효울성 테스트에서 실패한 나의 코드
def solution(participant, completion):
    for name in completion:
        if name in completion:
            participant.remove(name)
   
    answer=participant[0]
    return answer

효율성이 떨어지는 이유 

  1. 처음에 for문을 돌면서 completion안에 이름이 존재하는지 확인하고, remove 하는 과정에서 시간복잡도가 N^2으로 증가하는 상황.
  2. remove 내장함수 또한 일일이 리스트를 순회하며 원소를 찾아야 하기 때문에 효율성이 떨어질 수 밖에 없다. 

효율성 테스트에 성공한 최종 코드 
def solution(participant, completion):
    count_athlete={}
    answer=0
    for name in participant:
        if name in count_athlete.keys():count_athlete[name]+=1 
        else:count_athlete[name]=1
            
    for name in completion:
        if name in count_athlete.keys():count_athlete[name]+=1 
        else:count_athlete[name]=1 
            
    for key, value in count_athlete.items():
        if value%2!=0:
            answer=key
            break
    
    return answer

효율성 테스트 통과할 수 있었던 이유

  1. for문을 각 1회 도는 것은 시간복잡도가 N이므로 효율성과 상관이 없다. (상수는 무효화)
  2. dict 변수를 for문에서 선언할 때, 새로 선언할 경우 dict['name'] = 1 로 선언하고, 새로 선언하지 않는 경우 dict['name']+=1 로 이미 있는 key를 조회하는 것이 중요하다. 
  3. dict 변수를 for문으로 순회할 때 .items() 써주는 것 잊지 말기.
반응형