Условие
Return the k-th smallest value of the list (k counts from 1). Quickselect finds it in average O(n) time without sorting everything.
Implement kth_smallest(numbers, k).
Input
numbers: The list of integers. k: Which smallest value to return, starting at 1.
Output
Return the k-th smallest value.
Ограничения
1 ≤ k ≤ length ≤ 200,000
Примеры
Ввод:
k = 3
numbers = [7, 10, 4, 3, 20, 15]
Ожидаемый вывод:
expected = 7Ввод:
long long k = 3;
std::vector<long long> numbers = {7, 10, 4, 3, 20, 15};
Ожидаемый вывод:
long long expected = 7;Ввод:
long k = 3;
long[] numbers = new long[] {7, 10, 4, 3, 20, 15};
Ожидаемый вывод:
long expected = 7;Пояснение
Sorted: 3, 4, 7, 10, 15, 20; the 3rd smallest is 7.
Ввод:
k = 2
numbers = [5, 5, 1]
Ожидаемый вывод:
expected = 5Ввод:
long long k = 2;
std::vector<long long> numbers = {5, 5, 1};
Ожидаемый вывод:
long long expected = 5;Ввод:
long k = 2;
long[] numbers = new long[] {5, 5, 1};
Ожидаемый вывод:
long expected = 5;Пояснение
Sorted: 1, 5, 5; the 2nd smallest is 5 (duplicates count).
Как подступиться
Сначала попробуйте сами, затем открывайте подсказки по одной.
Подсказка 1
Нужен только элемент, который оказался бы на позиции k после сортировки, а не полностью отсортированный список.
Подсказка 2
Разбейте вокруг pivot и рекурсивно или итеративно идите только в ту сторону, где ещё может быть нужный ранг.
Подсказка 3
Помните, что k нумеруется с 1, и дубликаты тоже считаются отдельными позициями.
Ожидаемая сложность
O(n) average time, O(1) extra space
Ссылки
Это задание адаптировано для CodeUtility. Следуйте условию и ограничениям выше.