通过划分的方式在线性时间内找出一个序列中第K大的元素

#py2.7
#coding:utf-8
import random

def findkth(ll, k):
if k > len(ll):
raise Exception('Argument error')
else:
pivot = ll[random.randrange(0,len(ll))]
last = 0
for i in range(1, len(ll)):
if ll[i] >= pivot:
last += 1
swap(ll, i, last)
swap(ll, 0, last)
print(ll,k)
order = last+1
if order == k:
return pivot
elif order < k:
return findkth(ll[last + 1:], k - order)
else:
return findkth(ll[0:last], k)

def swap(ll, i, last):
tmp = ll[i]
ll[i] = ll[last]
ll[last] = tmp

ll = [1, 22, 31, 14, 0.5, 26, 17, 38, 19, 310]
k = 3
print(findkth(ll, k))