sort 함수가 런타임 에러를 발생시킬텐데.... 다른 애들은 그냥 반복이지 않나요...
그렇다고 sort 함수를 안쓸수도 없고 ...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 def solution(N, stages): a = [0]*(N+1) b = [0]*(N+1) for k in stages: if 1<=k<=N: a[k] += 1 for i in range(1, k+1): b[i] += 1 else: for j in range(1, len(b)): b[j] += 1 pre = [] for n in range(1, N+1): pre.append([a[n]/b[n], n]) pre.sort(key=lambda x: (-x[0], x[1])) answer = [] for p in pre: answer.append(p[1]) return answer
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
def solution(N, stages): a = [0]*(N+1) b = [0]*(N+1) for k in stages: if 1<=k<=N: a[k] += 1 for i in range(1, k+1): b[i] += 1 else: for j in range(1, len(b)): b[j] += 1 pre = [] for n in range(1, N+1): pre.append([a[n]/b[n], n]) pre.sort(key=lambda x: (-x[0], x[1])) answer = [] for p in pre: answer.append(p[1]) return answer
저도 같은 번호의 테스트가 런타임 에러나네요ㅠ
스테이지에 아무도 도달하지 못했을때 0으로 나누게 되어서 그렇네요;; 저도 한참을 고민했어요.
N = 4 stages = [1, 2, 3, 2, 1] return = [3, 2, 1, 4] 테스트케이스 추가해보세요~