[프로그래머스 level1] 실패율 - 파이썬

💡 문제 설명

슈퍼 게임 개발자 오렐리는 큰 고민에 빠졌다. 그녀가 만든 프렌즈 오천성이 대성공을 거뒀지만, 요즘 신규 사용자의 수가 급감한 것이다. 원인은 신규 사용자와 기존 사용자 사이에 스테이지 차이가 너무 큰 것이 문제였다.

이 문제를 어떻게 할까 고민 한 그녀는 동적으로 게임 시간을 늘려서 난이도를 조절하기로 했다. 역시 슈퍼 개발자라 대부분의 로직은 쉽게 구현했지만, 실패율을 구하는 부분에서 위기에 빠지고 말았다. 오렐리를 위해 실패율을 구하는 코드를 완성하라.

• 실패율은 다음과 같이 정의한다.
  • 스테이지에 도달했으나 아직 클리어하지 못한 플레이어의 수 / 스테이지에 도달한 플레이어 수

전체 스테이지의 개수 N, 게임을 이용하는 사용자가 현재 멈춰있는 스테이지의 번호가 담긴 배열 stages가 매개변수로 주어질 때, 실패율이 높은 스테이지부터 내림차순으로 스테이지의 번호가 담겨있는 배열을 return 하도록 solution 함수를 완성하라.

💡 제한사항

• 스테이지의 개수 N은 1 이상 500 이하의 자연수이다.
• stages의 길이는 1 이상 200,000 이하이다.
• stages에는 1 이상 N + 1 이하의 자연수가 담겨있다.
  • 각 자연수는 사용자가 현재 도전 중인 스테이지의 번호를 나타낸다.
  • 단, N + 1 은 마지막 스테이지(N 번째 스테이지) 까지 클리어 한 사용자를 나타낸다.
• 만약 실패율이 같은 스테이지가 있다면 작은 번호의 스테이지가 먼저 오도록 하면 된다.
• 스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다.

💡 나의코드

처음 생각해낸 방법은 1부터 N+1까지 for문을 돌리면서 stages 리스트에 i 마다 count 함수를 이용하여

stages.count(i) / len(stages) 해주는 방법을 생각했다. 

이와 같은 방법으로 구현을 한 후 각각의 실패율들을 새로운 리스트에 하나하나 집어넣었다

def solution(N, stages):
    People = len(stages)
    faillist = []
    for i in range(1, N + 1):
        faillist.append(stages.count(i) / People)
        if stages.count(i) / People > 0:
            People -= stages.count(i)

    print(faillist)


A = 5
B = [2, 1, 2, 6, 2, 4, 3, 3]
solution(A, B)

결괏값은 아래와 같이 각각의 실패율이 잘 저장된 것을 볼 수 있다.

하지만 우린 결과값을 실패율의 내림차순으로 배열의 인덱스를 출력해야 한다.

분명 for문을 한 번 더 돌리거나 하면 런타임 에러가 날 확률이 높을 것 같고, 더 좋은 방법이 있을 것만 같은 생각에 계속 고민을 했다.

그러던 중 아침에 다트 언어의 문법을 보면서 기억난 map!! 

💡 파이썬에서는 리스트를 딕셔너리로 대체해서 이용하면 풀 수 있을 것 같았다.!!

 

하지만 문제는 딕셔너리 값을 어떻게 전달했는지 기억이 나지 않는다 🤦‍♂️

그래서 딕셔너리 넣는 법을 다시 구글링.. 

 

간단했다. 리스트 [] 대신 딕셔너리인 {}로 선언을 해주고 '딕셔너리{인덱스} = 값'으로 넣으면 끝!

그리고 람다식을 이용하여 return을 해주었다.

💡 수정 코드

def solution(N, stages):
    People = len(stages)
    faillist = {}
    for i in range(1, N + 1):
        if People != 0:
            faillist[i] = stages.count(i) / People
            People -= stages.count(i)
        else:
            faillist[i] = 0

    return sorted(faillist, key=lambda i: faillist[i], reverse=True)

 

* 딕셔너리 잊지 말자

* 람다식 잊지 말자