题目描述
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
示例 1:
输入:nums = [2,0,2,1,1,0]
输出:[0,0,1,1,2,2]
示例 2:
输入:nums = [2,0,1]
输出:[0,1,2]
示例 3:
输入:nums = [0]
输出:[0]
示例 4:
输入:nums = [1]
输出:[1]
题解
数轴上,起始位置 i=0,j=0,k=n-1,三数排序要保证的是[0,j-1] 区间内的数是0,[j,i-1] 区间是1,[k,n-1] 区间是2。排序的过程如下:
- 如果nums[i]==0,就需要就将 i 指针所指向的数与 j 指针所指向的数交换,且由于交换过去了,j 已经是满足等于0的位置了,因此j++,i++
- 如果nums[i]==1,那么i++即可,要满足[j,i-1] 区间,因此i++
- 如果nums[i]==2,需要将2位置交换到k后面,因此k–,i 没有变化是因为交换过去的数字不确定是1还是0。
class Solution {
public:
void sortColors(vector<int>& nums) {
for(int i=0,j=0,k=nums.size()-1;i<=k;)
{
if(nums[i]==0) swap(nums[i++],nums[j++]);
else if(nums[i]==2) swap(nums[i],nums[k--]);
else if(nums[i]==1) i++;
}
}
};