‘큐에서 K개 꺼내기’은 큐의 핵심 연산을 다른 입력 조건에서도 스스로 선택할 수 있게 만드는 반복 문제입니다. 결과뿐 아니라 경계값과 목표 복잡도까지 확인하세요.
값을 큐에 넣고 앞에서 최대 k개를 꺼낸 리스트를 반환하세요.
take_first(values, k)
시간 O(n) 이내
실행하면 아래 순서대로 채점합니다. print() 출력도 이 순서대로 쌓입니다.
| # | 이름 | values | k | 기대값 |
|---|---|---|---|---|
| 1 | 두 개 | [1, 2, 3] | 2 | [1, 2] |
| 2 | 큰 k | [4] | 3 | [4] |
| 3 | 0 | [1, 2] | 0 | [] |
| 4 | 빈 큐 | [] | 2 | [] |
| 5 | k가 길이와 같음 | [-1, 0, 1] | 3 | [-1, 0, 1] |
Python 표준 라이브러리는 사용할 수 있습니다. 함수 이름과 매개변수는 제시된 형태를 유지하세요.
pop(0)은 O(n)이므로 collections.deque의 popleft()를 사용하세요.코드를 작성하고 실행하면 5개의 테스트가 각각 표시됩니다.