Pagini recente » Cod sursa (job #1739475) | Cod sursa (job #1778480) | Cod sursa (job #2537097)
import random
def nth_element(k, st, dr):
if st>=dr:
return
i = st
j = dr
a = random.choice(v[st:dr+1])
while i<=j:
while v[i]<a:
i += 1
while v[j]>a:
j -= 1
if i<=j:
v[i], v[j] = v[j], v[i]
i += 1
j -= 1
if k<=j-st+1:
nth_element(k, st, j)
else:
nth_element(k-(j-st+1),j+1,dr)
f = open("sdo.in", "r")
g = open("sdo.out", "w")
n, k = map(int,f.readline().split())
v = list(map(int,f.readline().split()))
nth_element(k, 0, len(v)-1)
g.write(str(v[k-1]))