실패율 (python)

https://school.programmers.co.kr/learn/courses/30/lessons/42889

프로그래머스

문제 설명

실패율

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

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

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

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

 

제한사항

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

---

code

# Lv.1
# 실패율
def solution(N, stages):
    failure_rate = {}
    stages_len = len(stages)

    for i in range(1, N+1):
        if stages_len != 0:
            stage_count = stages.count(i)
            failure_rate[i] = stage_count/stages_len # 실패율 계산
            stages_len -= stage_count # 다음 단계에 도달한 사람들 수
        else:
            failure_rate[i] = 0 # 사람이 없는 단계들은 일괄 처리

    answer = sorted(failure_rate.keys(), key=lambda key: failure_rate[key], reverse=True) # 실패율로 정렬

    return answer

 

알고리즘

    failure_rate = {}
    stages_len = len(stages)

 

for i in range(1, N+1):
        if stages_len != 0:
            stage_count = stages.count(i)
            failure_rate[i] = stage_count/stages_len # 실패율 계산
            stages_len -= stage_count # 다음 단계에 도달한 사람들 수
        else:
            failure_rate[i] = 0 # 사람이 없는 단계들은 일괄 처리

스테이지만큼 반복문을 돈다. 만약 stage_len 0이라면 사람이 없는 단계이기에, 해당 스테이지의 실패율 0을 failure_rate[스테지이]에 저장한다.

만약 stage_len이 0이 아니라면 해당 스테이지의 사람 수를 count를 통해 구해서 stage_count에 저장한다. 실패율의 정의에 따라 스테이지에 도달했으나 클리어하지 못한 수 / 스테이지에 도달한 사람 수를 failure_rate[스테이지]에 저장한다. 그 후 반복문을 통해 구할 다음 단계에는 도달하지 못한 사람이니 stages_len에서 stage_count를 빼서 저장한다.

 

위 반복문을 돌면 failure_rate에 key: 스테이지, value: 실패율이 기록된다.

    answer = sorted(failure_rate.keys(), key=lambda key: failure_rate[key], reverse=True) # 실패율로 정렬
    return answer

실패율 값을 기준으로 스테이지를 내림차순으로 정리해서 anwer에 저장한다. 그리고 답으로 반환한다.

 

피드백

# Lv.1
# 실패율
def solution(N, stages):
    # stage별 도달 참가자 수 구하기
    stage_score = [0] * (N + 2)
    for stage in stages:
        stage_score[stage] += 1
    # 각 stage별 실패율과 stage 저장하기
    stage_info = []
    for i in range(1, N + 1):
        stage_info.append([i, stage_score[i] / sum(stage_score[i:N + 2])])
    # 실패율을 기준으로 내림차순으로 정리
    stage_info.sort(key=lambda x:-x[1])
    return [i[0] for i in stage_info]

https://github.com/soaringwave/Python-algorithm-studying/commit/88bc6b02ecb26e289e6c7fab7805719deafcf5fa

처음엔 위처럼 코드륵 작성했으나, 코드 제출시 케이스 중 5개가 런타임 에러가 났다. 그래서 다른 사람의 풀이 중 가독성이 좋고, 간단한 코드로 피드백 했다.