def sort_three(nums):
    low = current = 0
    high = len(nums)
    while current < high:
        if nums[current] == 0:
            nums[low], nums[current] = nums[current], nums[low]
            low += 1
            current += 1
        elif nums[current] == 1:
            current += 1
        else:
            high -= 1
            nums[current], nums[high] = nums[high], nums[current]
    return nums