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

문제링크 : 프로그래머스 완주하지 못한 선수

 

파이썬을 공부하고 있어서 이에 익숙해지기 위하여 비교적 쉬운 난이도의 알고리즘 문제부터 접근을 하고 있다. 따라서 다른 사람의 풀이도 참고하기 위해 다른 사람의 풀이과정도 한번 살펴볼 예정이다.

 

문제

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

마라톤에 참여한 선수들의 이름이 담긴 배열 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

입출력 예 설명

• 예제 #1 : leo는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.
• 예제 #2 : vinko는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.
• 예제 #3 : mislav는 참여자 명단에는 두 명이 있지만, 완주자 명단에는 한 명밖에 없기 때문에 한명은 완주하지 못했습니다.

 

풀이

• 키&벨류가 한 쌍씩만 존재하기 때문에 이를 이용해 풀려고 접근을 시도함
• 근데 동명이인이 존재할 수 있음
• 따라서 새로운 딕셔너리를 만들고, 참가자 명단에 있는 이름을 키 값으로 주고 명수를 벨류로 설정하기로 함
• 예제 3을 예로 들었을 경우, dic[mislav] = 2, dic[stanko] = 1, dic[ana] = 1 이런식으로 초기 설정을 해주고
• completion에 있는 사람들을 -1 처리해준다.
• 그리고 마지막에 value가 0이 아닌 키 값을 리턴해준다.

 

코드

def solution(participant, completion):
    answer = ''
    dic = dict()

    for p in participant :
        if p not in dic.keys() :
            dic[p] = 1
        else:
            dic[p] += 1

    for c in completion :
        dic[c] -= 1

    for k in dic.keys() :
        if dic[k] != 0 :
            answer = k
            break
    
    return answer

참고할 다른 사람의 코드

# collections 모듈 사용한 사람들
import collections

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

# 정렬 사용한 사람들
def solution(participant, completion):
    participant.sort()
    completion.sort()
    for i in range(len(completion)):
        if participant[i] != completion[i]:
            return participant[i]
    return participant[len(participant)-1]

Collection 모듈을 공부해야겠다...