题目分析:
这一题是让我们对【0,1,2】三个元素进行排序,但是不能用内置的排序算法,而且要求最好用一趟扫描算法,要求在原数组上作出修改。
解题思路:
对数组元素逐一判断,如果是0就与数组第一个元素交换位置,如果是2就与数组最后一个元素交换位置,是1不做处理。
程序中设置了0元素结束位置,2元素开始位置,每次交换与标记的位置交换,不再与首个(最后一个)交换。当前位置大于2元素开始位置时退出循环。
测试代码: (Runtime: 36 ms, faster than 97.97%)
class Solution:
def sortColors(self, nums: list) -> None:
i = 0; start = 0; end = len(nums) - 1
while i <= end:
if nums[i] == 0:
nums[i], nums[start] = nums[start], nums[i]
start += 1
elif nums[i] == 2:
nums[i], nums[end] = nums[end], nums[i]
end -= 1
i -= 1
i += 1
print(nums) #仅做测试,提交时请删除该行
Solution().sortColors([2,0,2,1,1,0]) #仅做测试,提交时请删除该行