CodingTest/해시 알고리즘

프로그래머스 (Level 1) - 완주하지 못한 선수

seongduck 2022. 8. 6. 15:57

1. 문제 설명

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

  • 마라톤에 참여한 선수들의 이름이 담긴 배열 participant
  • 선수들의 이름이 담긴 배열 completion
  • 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성

2. 제한사항

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

3. 입출력 예

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"

 


4. 풀이 접근

  1. 참여 선수 명단 - 완주 선수 명단 (list 간의 뺄셈은 불가능)
  2. 딕셔너리로 만들고 참여 선수 명단 - 완주 선수 명단
  3. 각각 이름이 몇번 나왔는지 저장 후 Counter를 이용하여 뺄셈(Counter끼리는 뺄셈 가능)

5. 코드

import collections

def solution(participant, completion):
    par = collections.Counter(participant)
    com = collections.Counter(completion)
    
    result = par - com
    
    for key, value in result.items():
        answer = key
    
    return answer

6. 알아둘만한 문법

import collections

collections.Counter(a)
  • a리스트의 중복 포함 key, value를 나눠 저장가능
  • 리스트, 딕셔너리끼리 뺄셈, 덧셈 가능
  • value가 많은 값으로 내림차순

 

for key, value in result.items():
  • 원하는 key, value만 추출 가능

 

for i, name in enumerate(a)
  • 리스트에 있는 인덱스와 요소를 같이 꺼낼 때 사용하는 문법