循环程序中if和if-else

在看力扣上75题颜色分类时，题目中因为包含数字是0，1，2这三种类型。目的是遍历一遍，将三种类型数字放在指定位置。代码中if条件判断遇到这三种类型，因为只有三种类型数字，一开始自己写了三个if语句，认为三种情况就可以覆盖，结果运行时出错，代码如下：

在while语句中，虽然三个if语句对三种类型数字进行判断，但是while中第一个if语句中，对循环变量i进行了操作，之后会依次进入第二个和第三个if语句进行判断，导致最后程序没有达到想要的效果。

def sortColors(nums) -> None:
    """
    Do not return anything, modify nums in-place instead.
    """

    def swap(nums, p1, p2):
        nums[p1], nums[p2] = nums[p2], nums[p1]

    if len(nums) < 2: return nums
    # 开始时候，三个区间重置为空
    zeros = 0
    i = 0
    two = len(nums)
    while i < two:
        if nums[i] == 0:
            swap(nums, i, zeros)
            i += 1
            zeros += 1
        if nums[i] == 1:
            i += 1
        if nums[i] == 2:
            two -= 1
            swap(nums,