思路:对三种颜色分别进行计数,再对原来的数组进行覆盖写的操作。
class Solution(object):
# def sortColors(self, nums): # 直接调用了库函数sort()
# """
# :type nums: List[int]
# :rtype: None Do not return anything, modify nums in-place instead.
# """
# nums.sort()
def sortColors(self, nums):
count_0 = 0
count_1 = 0
count_2 = 0
for num in nums:
if num == 0:
count_0 += 1
elif num == 1:
count_1 += 1
elif num == 2:
count_2 += 1
for i in range(len(nums)): # 修改原来的数组
if i < count_0:
nums[i] = 0
if i >= count_0 and i < count_0+count_1:
nums[i] = 1
if i >= count_0 + count_1:
nums[i] = 2