K-th Smallest Element

Problema

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).

numbers: The list of integers. k: Which smallest value to return, starting at 1.

Return the k-th smallest value.

1 ≤ k ≤ length ≤ 200,000

Exemplo 1

Entrada:
k = 3
numbers = [7, 10, 4, 3, 20, 15]

Saída esperada:
expected = 7

Entrada:
long long k = 3;
std::vector<long long> numbers = {7, 10, 4, 3, 20, 15};

Saída esperada:
long long expected = 7;

Entrada:
long k = 3;
long[] numbers = new long[] {7, 10, 4, 3, 20, 15};

Saída esperada:
long expected = 7;

Sorted: 3, 4, 7, 10, 15, 20; the 3rd smallest is 7.

Exemplo 2

Entrada:
k = 2
numbers = [5, 5, 1]

Saída esperada:
expected = 5

Entrada:
long long k = 2;
std::vector<long long> numbers = {5, 5, 1};

Saída esperada:
long long expected = 5;

Entrada:
long k = 2;
long[] numbers = new long[] {5, 5, 1};

Saída esperada:
long expected = 5;

Sorted: 1, 5, 5; the 2nd smallest is 5 (duplicates count).

Tente sozinho primeiro e depois abra uma dica de cada vez.

Dica 1

Você só precisa do elemento que ficaria na posição k após ordenar, não da lista inteira ordenada.

Dica 2

Particione em torno de um pivô e recorra ou itere apenas para o lado que ainda pode conter a posição alvo.

Dica 3

Lembre que k é baseado em 1, e duplicatas ainda contam como posições separadas.

Complexidade esperada

O(n) average time, O(1) extra space

Este desafio foi adaptado para o CodeUtility. Siga o enunciado e as restrições acima.