LeetCode0075.颜色分类 Go语言AC笔记

时间复杂度：O(n)

解题思路

一开始想的是双指针（是真正意义上的双指针，不是官方题解所谓的“双指针”，我更感觉官方题解那是三指针），但是如果遇到两指针都是1的情况解决起来会很麻烦，但不是解决不了。看了官方题解的“双指针”，醍醐灌顶，颇有收获。

首先我们定义好p0和p2两个指针，前者在头部，后者在尾部，分别指向已经排好序的0的后一个下标和已经排好序的2的前一个下标，需要格外注意p2指针的定义，我的这个定义和官方题解的p2定义并不一样。

接下来我们就是要确定p0和p2两个指针如何移动。我们开始从头遍历数组（正因需要遍历数组而不是单纯依靠p0和p2两个指针，我认为该解题思路是三指针做法而不是双指针），如果当前元素为0，那么就交换p0元素和当前元素，并根据p0指针定义让p0右移。同理如果当前元素为2，那么就交换当前元素和p2的元素，并根据p2指针定义让p2右移。当遍历到p2时，结束遍历。

看上去这个过程很简单，但实际上我们需要注意两处细节。

一是我们务必要保证p2指针指向的是已经排好序的2的头部前一个下标。所以在p2左移过程中如果p2的元素为2，那么我们要让p2继续左移，直至p2的元素不是2为止。具体原因请参考官方题解，简单说来就是如果p2的元素不是2那么我们会将2换到遍历的位置，并且将其略过遍历下一个位置。

二是如果遍历元素为2交换后换到了0，那么还需要进行0的交换。该点在代码的体现就是不要写else if，而是两个并列if先判断遍历元素是否为2，接下来还要判断遍历元素是否为0。

AC代码

func sortColors(nums []int)  {
    p0,p2:=0,len(nums)-1
    for i:=0;i<=p2;{
        //保证p2指针的定义
        if nums[p2]==2{
            p2--
            continue
        }
        if nums[i]==2{
            nums[i],nums[p2]=nums[p2],nums[i]
            p2--
        }
        //千万不要写成else if，因为我们可能交换后获得了0
        if nums[i]==0{
            nums[p0],nums[i]=nums[i],nums[p0]
            p0++
        }
        i++
    }
}

感悟

此题技巧性很强，并且需要注意细节。