leetcode之Sort Colors

class Solution(object):
def sortColors(self, nums):
"""
:type nums: List[int]
:rtype: void Do not return anything, modify nums in-place instead.
"""
if len(nums) == 1:
return
if len(nums) == 2:
if nums[0] == 0:
return
elif nums[-1] == 2:
return
else:
a = nums.pop()
nums.insert(0, a)
LocationOfRed = -1
LocationOfWhite = -1
#此处多加了一个blue
nums.append(2)
for i in range(len(nums) - 1):
if nums[0] == 0:
nums.insert(LocationOfRed, 0)
del nums[0]
elif nums[0] == 1:
nums.insert(LocationOfWhite, 1)
del nums[0]
LocationOfRed = LocationOfRed - 1
else:
nums.append(2)
del nums[0]
LocationOfRed = LocationOfRed - 1
LocationOfWhite = LocationOfWhite - 1
nums.pop()