|
안녕하세요. 조카가 며칠째 머리를 싸메고 있어서 이곳에 질문드려봅니다.
• 입력:
• n개의 서로 다른 정수 값 (리스트 A에 저장한다고 가정한다)
• 3<=n <= 100,000
• 출력:
• mli] 값을 A[01,... AI] 중 (//3 + 1)-번째로 작은 값으로 정의한다
• 제일 작은 값은 1번째로 작다고 정의하다
• 모든 1 = 0..., 7-1에 대해, m[i] 값의 합을 출력한다
• 여러 알고리즘이 존재한다. 수행 시간이 가능하면 작도록 알고리즘을 설계하시오
• 당연하지만, 정렬(sort)을 사용해도 안됨!
===========================
import heapq as h
A = [int(k) for k in input().split()]
sum = 0
for k in range(len(A)):
sum += h.nsmallest(k // 3 + 1, A[ : k + 1])[k // 3]
print(sum)
위 처럼 작성했고 코드는 도는데 10초 이상 걸린다고 하네요.
힌트 좀 주시기 바랍니다.
감사합니다.
| 
등록안내
등록안내
문제가 뭐라고 되어있는지 이해가 안되네요.
스크린샷은 따로 없나요?