[프로그래머스] 실패율

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

프로그래머스

---

처음에 제출했었던 코드 (점수:70점)

def solution(N, stages):
    answer = []
    
    failed=[0]*501 #각 스테이지에 멈춰있는 사용자 수 저장할 배열
    cleared=[0]*501 #각 스테이지에 도달한 사용자 수 저장할 배열
    
    for i in stages:
        failed[i] += 1
        for j in range(1,i+1):
            cleared[j] += 1
            
    arr=[]
    for i in range(1,501):
    	#분모가 0이 되지 않는 경우에만 실패율과 스테이지번호 저장
        if cleared[i] != 0:
            arr.append((failed[i]/cleared[i],i))
    
    #첫번째조건: 실패율 내림차순, 두번째조건: 스테이지번호 오름차순으로 정렬
    arr.sort(key=lambda x:(-x[0],x[1]))
    
    #정렬한 스테이지 번호를 순서대로 배열에 담아 리턴
    for rate,num in arr:
        if num<=N:
            answer.append(num)
    return answer

 

테스트케이스 중 몇몇 개에서만 오답이 떴다.

 

N에 도달한 사용자가 한 명도 없을 경우 실패율 0으로 arr에 append되어야 하는데,

이 경우를 고려하지 않아서 오답이 떴던 거였다.

 

예를 들어 인풋 N=4, stages=[1,1,2,3,3,3]일 때

아웃풋으로는 [3,2,1,4] 가 되어야 하는데

 

위 코드상으로는 [3,2,1] 만 리턴되는 오류가 있었다.

 

이러한 점을 고려하여 코드를 수정하니까 정답 판정을 받았다.

---

정답 코드

def solution(N, stages):
    answer = []
    
    failed=[0]*502 #각 스테이지에 멈춰있는 사용자 수
    cleared=[0]*502 #각 스테이지에 도달한 사용자 수
    
    for i in stages:
        failed[i] += 1
        for j in range(1,i+1):
            cleared[j] += 1
            
    arr=[]
    for i in range(1,N+1):
    	#분모가 0이 되지 않는 경우, 실패율과 스테이지 번호 저장
        if cleared[i] != 0:
            arr.append((failed[i]/cleared[i],i))
        #해당 스테이지까지 도달한 사용자 없는 경우,
         실패율은 0으로 저장
        else:
            arr.append((0,i))
    
    #첫번째조건:실패율 내림차순, 두번째조건:스테이지번호 오름차순으로 정렬
    arr.sort(key=lambda x:(-x[0],x[1]))
    
    #정렬한 스테이지 번호를 순서대로 배열에 담아 리턴
    for rate,num in arr:
        if num<=N:
            answer.append(num)
    return answer

 

+추가

   1. 스테이지 개수 N은 1이상 500이하의 자연수랬고
   2. stages에는 1이상 N+1 이하의 자연수가 담겨있다고 한다.

 

처음에는 1번 조건만을 보고 failed,cleared 리스트의 크기를 501로 설정했었는데,

테스트케이스 중 2개에서 런타임 에러가 떴다.

 

arr1, arr2에 값을 저장해주는 과정에서 (line 7~10)

stages 값 중에 정확히 N+1이 있는 경우 인덱스 에러가 떠서 그런 것 같다.

 

그래서 두 리스트의 크기를 502로 설정해주니 모든 테스트케이스에서 실패 없이 잘 실행되었다.