Programmers - 실패율

스테이지의 개수 N, 게임을 이용하는 사용자가 현재 멈춰있는 스테이지의 번호가 담긴 배열 stages가 매개변수로 주어질 때, 실패율이 높은 스테이지부터 내림차순으로 스테이지의 번호가 담겨있는 배열을 만드는 문제이다.

 

사용한 언어는 Python3이다.

사용자가 멈춰있는 스테이지를 구하고, 해당 스테이지를 도전한 사용자와 멈춰있는 사용자의 수를 계산하여 실패율을 구한다. 처음 result를 만들 때 모든 스테이지를 통과한 사용자는 제외하기 위해 스테이지 1부터 스테이지 5까지만 범위로 잡았다.

코딩테스트 연습 - 실패율

def solution(N, stages):
    result = [[i for i, x in enumerate(stages) if x == j] for j in range(1, N+1)]
    result_2 = []; steps = 0
    for i, j in enumerate(result):
        if len(stages) - steps != 0:
            result_2.append((i+1, len(j)/(len(stages) - steps))) 
            steps += len(j)
        else : result_2.append((i+1, 0))
    answer = [i[0] for i in sorted(result_2, key = lambda x: x[1], reverse = True)]
    return answer