143. 排颜色 II

class Solution:
    """
    @param colors: A list of integer
    @param k: An integer
    @return: nothing
    """
    def sortColors2(self, colors, k):
        # write your code here
        #一层一层往下划分，小于k的放置左边，大于k的放置右边，left会随着k更新
        #背向型双指针
        
        length = len(colors)
        #left只需要重置第一次即可，后面的会更新，一直不停的缩小区间，来进行划分数组，往右边走
        left = 0
        index = 0
        
        #最外层
        for i in range(1, k + 1):
　　　　　　　#right每次需要重置，left不需要，left和index为起点指向，index正常走，left记录小于i的个数，缩小区间.
            right = length - 1
            
            #里面不停的进行划分
            while index <= right:
                #放置左边
                if (colors[index] < i):
                    colors[index], colors[left] = colors[left], colors[index]
                    index += 1 
                    left += 1 
                elif (colors[index] > i):
                    colors[index], colors[right] = colors[right], colors[index]
                    right -= 1 
                else:
                    index += 1 
            
            #此时已经划分完毕当前i的，如果是小于i的左边，大于i的右边，只需要关心大于i的部分即可，更新index
            index = left 
        
        return colors