075 Sort Colors

class Solution:
    # @param {integer[]} nums
    # @return {void} Do not return anything, modify nums in-place instead.
    def sortColors(self, nums):
        start, end = 0, len(nums) - 1
        while start < len(nums) and nums[start] == 0:
            start += 1
        k = start
        while start < end:
            while end > start and nums[end] == 2:
                end -= 1
            while start < end and nums[start] == 0:
                start += 1
            if k > end:
                break
            if nums[k] == 2:
                nums[k], nums[end] = nums[end], nums[k]
                end -= 1
            elif nums[k] == 0:
                nums[k], nums[start] = nums[start], nums[k]
                start += 1
            else:
                k += 1