Update quick_sort.py

Streamlining quick_sort function
aravindiiitb · Aug 29, 2016 · 66433a5 · 66433a5
1 parent abd9d78
commit 66433a5
Showing 1 changed file with 11 additions and 12 deletions.
diff --git a/sorts/quick_sort.py b/sorts/quick_sort.py
@@ -35,21 +35,20 @@ def quick_sort(collection):
     >>> quick_sort([-2, -5, -45])
     [-45, -5, -2]
     """
+    if len(collection) <= 1:
+        return collection
     less = []
     equal = []
     greater = []
-    if len(collection) > 1:
-        pivot = collection[0]
-        for x in collection:
-            if x < pivot:
-                less.append(x)
-            if x == pivot:
-                equal.append(x)
-            if x > pivot:
-                greater.append(x)
-        return quick_sort(less) + equal + quick_sort(greater)
-    else:
-        return collection
+    pivot = collection[0]
+    for x in collection:
+        if x < pivot:
+            less.append(x)
+        elif x == pivot:
+            equal.append(x)
+        else:
+            greater.append(x)
+    return quick_sort(less) + equal + quick_sort(greater)
 
 
 if __name__ == '__main__':