[programmers python] 42889 실패율

 

 

2019 KAKAO BLIND RECRUITMENT

https://programmers.co.kr/learn/courses/30/lessons/42889?language=python3 

코딩테스트 연습 - 실패율

 

 

[문제]

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

 

[문제설명]

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

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

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

 

[제한조건]

 

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

 

---

 

[예시]

 

N	stages	result
5	[2, 1, 2, 6, 2, 4, 3, 3]	[3,4,2,1,5]
4	[4,4,4,4,4]	[4,1,2,3]

 

[예시 설명]

입출력 예 #1
1번 스테이지에는 총 8명의 사용자가 도전했으며, 이 중 1명의 사용자가 아직 클리어하지 못했다. 따라서 1번 스테이지의 실패율은 다음과 같다.

• 1 번 스테이지 실패율 : 1/8

2번 스테이지에는 총 7명의 사용자가 도전했으며, 이 중 3명의 사용자가 아직 클리어하지 못했다. 따라서 2번 스테이지의 실패율은 다음과 같다.

• 2 번 스테이지 실패율 : 3/7

마찬가지로 나머지 스테이지의 실패율은 다음과 같다.

• 3 번 스테이지 실패율 : 2/4
• 4번 스테이지 실패율 : 1/2
• 5번 스테이지 실패율 : 0/1

각 스테이지의 번호를 실패율의 내림차순으로 정렬하면 다음과 같다.

• [3,4,2,1,5]

 

입출력 예 #2

모든 사용자가 마지막 스테이지에 있으므로 4번 스테이지의 실패율은 1이며 나머지 스테이지의 실패율은 0이다.

• [4,1,2,3]

 

---

 

첫 번째 시도 _ 실패

 

단순하게 생각해보았다!

1. stages개수 구해서 users에 저장
2. for문 통해 users != 0:
   1. stages에서 i인 것 counting
   2. 실패율 연산
   3. result[i] = 실패율
   4. users-=count
3. 아니라면 result[i] = 0
4. sorted하여 return

 

def solution(N, stages):
    result = []
    users = len(stages)
    
    for i in range(1, N+1):
        if users != 0:
            count = stages.count(i)
            result.append(count/users)
            users -= count
        else:
            result.append(0)
    return sorted(result, reverse = True)

 

---

 

두 번째 시도 _ 성공

 

바보같은 아이디어로 ,, 이전에 단계들을 출력하는 것이 아니라 실패율을 출력하도록 설정해놓았다..

그 부분 수정!!

 

def solution(N, stages):
    result = {}
    users = len(stages)
    
    for i in range(1, N+1):
        if users != 0:
            count = stages.count(i)
            result[i] = (count/users)
            users -= count
        else:
            result[i] = 0
    return sorted(result, key=lambda x: result[x], reverse = True)

 

 

---

 

세 번째 시도 _ 성공

 

count함수를 안쓰는 방법도 해보았지만, 시간은 이전 것이 더 효율적이었다 ,, 

 

def solution(N, stages):
    result = {}
    users = len(stages)
    
    for i in range(1, N+1):
        count = 0
        if users != 0:
            for stage in stages:
                if stage == i:
                    count += 1
            result[i] = (count/users)
            users -= count
        else:
            result[i] = 0
    return sorted(result, key=lambda x: result[x], reverse = True)

 

 

---

 

네 번째 시도 _ 성공

 

count 함수 사용하면 생각보다 시간이 많이 나와서 defaultdict 활용하여 초기화 시킨 후 직접 counting 하는 코드로 수정해봄

 

from collections import defaultdict

def solution(N, stages):
    stage_dict = defaultdict(int)
    for stage in stages:
        stage_dict[stage] += 1
        
    result = {}
    users = len(stages)
    for i in range(1, N+1):
        users -= stage_dict[i-1]
        result[i] = stage_dict[i]/users if users else 0
    return sorted(result, key=lambda x: result[x], reverse = True)

 

 

---

시간이 훨씬 줄어든 것 확인 가능

함수 활용이 무조건 좋은 것만은 아닌 듯 함 ,, !

 

---

 

 

최종 코드