From 022533dc4a7d76a4865c797b404addf090fe9c54 Mon Sep 17 00:00:00 2001 From: Danila Fedorin Date: Fri, 27 Sep 2019 14:35:28 -0700 Subject: [PATCH] Fix index off by 1 --- qselect.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)