时间复杂度: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++
}
}
感悟
此题技巧性很强,并且需要注意细节。