[프로그래머스] 실패율 (Python)

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

프로그래머스

 

 

💬 시간초과 받은 코드

- 처음에는 스테이지별로 도달한 사람과 도전 중인 사람을 각각 arrival과 challenge 배열에 기록하고, 딕셔너리 형태로 실패율을 구해주었다.

- 그리고 이를 배열로 변환하여 조건에 맞게 실패율 내림차순, 스테이지 숫자 오름차순으로 정렬한 후 answer에 스테이지 숫자만 append 해주었다.

- 그러나 시간초과를 받았다ㅠㅠ

def solution(N, stages):
    answer = []
    arrival = [0]*(N+1)
    challenge = [0]*(N+1)
    fail = {}

    # stages를 순회하면서 도달한 스테이지와 현재 도전 중인 스테이지를 카운트해준다.
    for stage in stages:
        # stage가 N+1일 경우 limit을 stage로 둔다 => 1부터 N까지 체크
        if stage == N+1:
            limit = stage
        # 그 외의 경우, limit을 stage+1로 둔다 => 1부터 stage까지 체크
        else:
            limit = stage+1

        for num in range(1, limit):
            arrival[num] += 1

        # 모든 스테이지를 통과한 경우가 아니라면 스테이지 별로 도전 중인 유저 카운트
        if stage != N+1:
            challenge[stage] += 1

    # 실패율 구하기 => 도전자 / 도달자
    for num in range(1, N+1):
        fail[num] = challenge[num] / arrival[num]

    # 객체를 배열로 변환
    fail_array = list(fail.items())

    # 실패율 순으로 내림차순, 스테이지 숫자 순으로 오름차순 정렬
    fail_array.sort(key=lambda x : (-x[1], x[0]))

    # 스테이지 숫자만 차례로 answer에 append
    for stage_num in fail_array:
        answer.append(stage_num[0])

    return answer

 

 

💬 통과한 코드

- 생각과 구글링을 통해 로직을 수정해보았다. 유저 수(stages의 길이)를 length로 선언하고, 1부터 N까지 각 스테이지의 실패율을 바로 구해주었다. 그리고 실패율이 0을 넘는다면, 유저 수에서 도전 중인 사람 수를 빼주었다.

- 그 후 실패율을 담은 딕셔너리를 배열로 변환하고, 정렬 후 스테이지 숫자만 answer에 append 해주는 것은 위의 로직과 동일하다.

def solution(N, stages):
    answer = []
    fail = {}
    length = len(stages)

    # 실패율이 0 초과이면 length에서 해당 단계에 도전 중인 사람 수 빼기
    for num in range(1, N+1):
        if length != 0:
            rate = stages.count(num) / length
            fail[num] = rate
            if rate > 0:
                length -= stages.count(num)
        else:
            fail[num] = 0

    # 객체를 배열로 전환한 후, 실패율 내림차순 & 스테이지 숫자 오름차순 정렬
    fail_array = list(fail.items())
    fail_array.sort(key=lambda x:(-x[1], x[0]))

    # 스테이지 숫자만 answer에 append
    for item in fail_array:
        answer.append(item[0])

    return answer