[프로그래머스] Lv 1. 실패율 ( js, python)

🔅문제🔅

2019 KAKAO BLIND RECRUITMENT - 실패율

입력값 

N : 전체 스테이지의 수 ex) 5

stages : 게임 이용자가 현재 멈춰있는 스테이지의 번호들의 배열 ex) [2,1,2,6,2,4,3,3]

 

실패율  : 스테이징 도달했으나 아직 클리어 못한 플레이어 수 / 스테이징에 도달한 플레이어 수 

 

출력값

: 각 스테이지의 실패율이 큰 순서대로 스테이지 번호를 배열 형태로 출력하라 ex)[3,4,2,1,5]

 

👾해결과정👾

1. for문을 1부터 N까지 돌리면서 인덱스를 스테이지 번호로 이용해야겠다. 

2. 실패율을 구하는 분모와 분자를 두고 1~N까지의 실패율을 구하기 위해 이중 for문 

- 분모 : 스테이지 번호보다 크거나 같은 수, 분자  : 해당 숫자 

3. 실패율에 따른 인덱스 번호를 내림차순으로 정렬하고 인덱스 번호만 나오게 출력해야겠다.

 

✨최종코드✨

function solution(N, stages) {
    var answer = [];
    for( i = 1; i <= N; i++){
        var a = 0; //분모: 스테이지에 도달한 수
        var b = 0; //분자: 도달했지만 클리어 못한 숫자 == 해당 숫자
        for( j = 0; j < stages.length; j++){
            if(stages[j] >= i){
                a++;
            }
            if(stages[j] == i){
                b++;
            }
        }
        answer.push([i, b/a]);
        answer.sort((a,b) => b[1] - a[1]);
        var result = answer.map((num) => num[0]);  
    }
    return result;
}

 

+파이썬

+더 효율적으로

 

python3

def solution(N, stages):
    answer = []
    for i in range(1, N+1):
        a,b = 0,0
        for j in stages:
            if i <= j:
                b += 1
            if i == j:
                a += 1
        answer.append((i,a/b))
    answer.sort(key = lambda x:-x[1])
    return [x[0] for x in answer]

 이렇게 하니까 시간초과가 떠서 이중 for문 말고 다른 방법으로 해야 한다. 

 

def solution(N, stages):
    answer = []
    b = len(stages) # 분모
    for i in range(1, N+1):
        if b != 0:
            a = stages.count(i) #분자
            answer.append((i , a/b))
            b -= a 
        else:
            answer.append((i, 0))
    answer.sort(key = lambda x:-x[1])
    return [x[0] for x in answer]

이렇게 하면 통과된다!