1. 题⽬链接:75.颜⾊分类
2. 题⽬描述:
3. 解法(快排思想-三指针法使数组分三块):
算法思路:
类⽐数组分两块的算法思想,这⾥是将数组分成三块,那么我们可以再添加⼀个指针,实现数组分 三块。
设数组⼤⼩为n ,定义三个指针left, cur, right :
◦ left :⽤来标记0 序列的末尾,因此初始化为-1 ;
◦ cur :⽤来扫描数组,初始化为0 ;
◦ right :⽤来标记2 序列的起始位置,因此初始化为n 。
在cur 往后扫描的过程中,保证:
◦ [0, left] 内的元素都是0 ;
◦ [left + 1, cur - 1] 内的元素都是1 ;
◦ [cur, right - 1] 内的元素是待定元素;
◦ [right, n] 内的元素都是2 。
算法流程:
a. 初始化cur = 0,left = -1, right = numsSize ;
b. 当cur < right 的时候(因为right 表⽰的是2 序列的左边界,因此当cur 碰到right 的时候,说明已经将所有数据扫描完毕了),⼀直进⾏下⾯循环:根据nums[cur] 的值,可以分为下⾯三种情况:
i. nums[cur] == 0 ;说明此时这个位置的元素需要在left + 1 的位置上,因此交 换left + 1 与cur 位置的元素,并且让left++ (指向0 序列的右边界), cur++ (为什么可以 ++ 呢,是因为left + 1 位置要么是0 ,要么是cur ,交换 完毕之后,这个位置的值已经符合我们的要求,因此cur++ );
ii. nums[cur] == 1 ;说明这个位置应该在left 和cur 之间,此时⽆需交换,直接 让cur++ ,判断下⼀个元素即可;
iii. nums[cur] == 2 ;说明这个位置的元素应该在right - 1 的位置,因此交换right - 1 与cur 位置的元素,并且让right-- (指向2 序列的左边界), cur 不变(因为交换过来的数是没有被判断过的,因此需要在下轮循环中判断)
c. 当循环结束之后: [0, left] 表⽰0 序列; [left + 1, right - 1] 表⽰1 序列;[right, numsSize - 1] 表⽰2 序列。
C++算法代码:
class Solution
{
public:
void sortColors(vector<int>& nums)
{
int left=-1,right=nums.size(),i=0;
//left之前为0,right之后为2,中间遍历结束后完成分类
while(i < right)
{
if(nums[i] == 0)
{
swap(nums[++left], nums[i++]);
}
else if(nums[i] == 1)
{
i++;
}
else
{
swap(nums[--right], nums[i]);
}
}
}
};
Java算法代码:
class Solution {
public void swap(int[] nums, int i, int j)
{
int t = nums[i];
nums[i] = nums[j];
nums[j] = t;
}
public void sortColors(int[] nums) {
int left = -1, right = nums.length, i = 0;
while (i < right)
{
if (nums[i] == 0) swap(nums, ++left, i++);
else if (nums[i] == 1) i++;
else swap(nums, --right, i);
}
}
}