[프로그래머스] 42576 완주하지 못한 선수 (+딕셔너리 get) (python 파이썬)

프로그래머스

---

문제 난이도는 쉽지만, 익혀두면 좋은 기법들이 많다. 파이썬을 주력으로 쓰는 만큼 리스트 컴프리헨션을 많이 활용해야겠다.

---

아이디어

participant에 있는 이름을 key, 이름의 중복 개수를 value로 딕셔너리를 만들어준다.

 

그리고 completion를 순회하며 이름이 나올 때마다 -1 해준다. 

 

문제에서 한 명의 선수만 완주하지 못했다고 했으므로 딕셔너리에서 value의 값이 1인 걸 찾아 return 해주면 된다.

---

전체 코드

def solution(participant, completion):
    
    dic = {}
    
    for p in participant:
        dic[p] = dic.get(p, 0) + 1
    
    for c in completion:
        dic[c] -= 1
    
    dnf = [k for k, v in dic.items() if v > 0]
    
    return dnf[0]

---

코드 설명

for p in participant:
	dic[p] = dic.get(p, 0) + 1

 

딕셔너리에 get메서드를 사용하면 깔끔하게 key, value 쌍을 추가할 수 있다.

dic.get(x)는 하면 key가 x인 value를 반환해 준다. 

여기서 dic.get(x, '디폴트 값')는, dic에서 x의 대응되는 key가 없는 경우 디폴트 값을 반환해 준다.

 

따라서 dic[p] = dic.get(p, 0) + 1 는 dic안에 p라는 key가 있으면 value + 1, p라는 key가 없으면 dic[p] = 1이 된다.

이런 방식을 사용하면 if문을 사용하지 않아도 깔끔하게 구현 가능하다.

 

    dnf = [k for k, v in dic.items() if v > 0]
    
    return dnf[0]

딕셔너리에 items라는 메서드와 리스트 컨프리헨션을 사용해서 value가 0보다 큰 경우, key와 value를 반환해 준다.

문제 조건에서 완주 못한 선수는 1명이므로 dnf[0]으로 선수 이름을 반환해 주면 된다.