Update sort-even-and-odd-indices-independently.py

Author: kamyu104

Python/sort-even-and-odd-indices-independently.py

@@ -10,10 +10,10 @@ def sortEvenOdd(self, nums):
         """
         def partition(index, nums):
             for i in xrange(len(nums)):
-                j = index(i)
-                while nums[i] > 0:
-                    nums[i], nums[j] = nums[j], ~nums[i]  # processed
+                j = i
+                while nums[i] >= 0:
                     j = index(j)
+                    nums[i], nums[j] = nums[j], ~nums[i]  # processed
             for i in xrange(len(nums)):
                 nums[i] = ~nums[i]  # restore values
 
@@ -26,13 +26,10 @@ def inplace_counting_sort(nums, left, right, reverse=False):  # Time: O(n)
             for i in xrange(1, len(count)):
                 count[i] += count[i-1]
             for i in reversed(xrange(left, right+1)):  # inplace but unstable sort
-                if nums[i] < 0:  # processed
-                    continue
-                while i != left+count[nums[i]]-1:
+                while nums[i] >= 0:
                     count[nums[i]] -= 1
-                    nums[left+count[nums[i]]], nums[i] = ~nums[i], nums[left+count[nums[i]]]
-                count[nums[i]] -= 1
-                nums[i] = ~nums[i]
+                    j = left+count[nums[i]]
+                    nums[i], nums[j] = nums[j], ~nums[i]
             for i in xrange(left, right+1):
                 nums[i] = ~nums[i]  # restore values
             if reverse:  # unstable
@@ -59,10 +56,10 @@ def sortEvenOdd(self, nums):
         """
         def partition(index, nums):
             for i in xrange(len(nums)):
-                j = index(i)
-                while nums[i] > 0:
-                    nums[i], nums[j] = nums[j], ~nums[i]  # processed
+                j = i
+                while nums[i] >= 0:
                     j = index(j)
+                    nums[i], nums[j] = nums[j], ~nums[i]  # processed
             for i in xrange(len(nums)):
                 nums[i] = ~nums[i]  # restore values