본문 바로가기

알고리즘/프로그래머스(Python)

[알고리즘] 프로그래머스 완주하지 못한 선수 / python

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

 

코딩테스트 연습 - 완주하지 못한 선수 | 프로그래머스

수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다. 마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요. 제한사항 마라톤 경기에 참여한 선수의 수는 1명 이상 100,000명 이하입니다. completion의 길이는 partic

programmers.co.kr

 

참여자가 숫자로 주어졌을땐, 배열의 인덱스를 이용하면 된다

문자열으로 주어졌을땐 어떻게 해야할까?

더해서, 제한사항에 동명이인이 있을 수 있다는 조건이 걸렸다

이럴때는 해쉬 즉, 파이썬의 dictionary를 이용한다

참여자 leo 1 / kiki 1 / eden 1

완주자 eden 0 / kiki 0

value값이 1인 key의 값을 return 해주면 된다

 

def solution(participant, completion):
    answer = ''
    dic = {}
    
    for i in participant:
        if i in dic.keys():
            dic[i] += 1
        else:    
            dic[i] = 1
        
    for j in completion:
        dic[j] -= 1
    
    for key in dic:
        if dic[key] == 1:
            answer = key
    
    return answer

 

정석코드

def solution(participant, completion):
	d = {}
    
    for x in participant:
    	d[x] = d.get(x, 0) + 1
        
    for x in completion:
    	d[x] -= 1
    
    dnf = [k for k, v in d.items() if v > 0]
    answer = dnf[0]
    
    return answer
d.get(x, 0) + 1  # x가 딕셔너리에 있다면 (x의 value + 1)값을 반환, 없다면 (0 + 1)값을 반환