Pagini recente » Cod sursa (job #2808629) | Cod sursa (job #1904279) | Cod sursa (job #3243895) | Cod sursa (job #2528193) | Cod sursa (job #2536821)
import random
def kth_element(a : list, k : int):
x = random.choice(a)
to_left = [y for y in a if y < x]
to_right = [y for y in a if y > x]
if len(to_left) >= k:
return kth_element(to_left, k)
if k <= len(a) - len(to_right):
return x
return kth_element(to_right, k - (len(a) - len(to_right)))
f = open('sdo.in', 'r')
n, k = [int(x) for x in f.readline().split()]
lst = [int(x) for x in f.readline().split()]
g = open('sdo.out', 'w')
g.write(str(kth_element(lst, k)))
f.close()
g.close()