diff --git a/qselect.py b/qselect.py index fdb3abe..424c594 100644 --- a/qselect.py +++ b/qselect.py @@ -7,9 +7,9 @@ def qselect(i, xs): left = [x for x in xs if x < pivot] right = [x for x in xs if x >= pivot] - if i > len(left): + if i > len(left) + 1: return qselect(i - len(left) - 1, right) - elif i == len(left): + elif i == len(left) + 1: return pivot else: return qselect(i, left)