题目
给定一个包含红色、白色和蓝色,一共 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]
题解
class Solution {
public void sortColors(int[] nums) {
int len=nums.length;
//快速排序
//[0,zeros),[zeros,two),[two,len-1]一开始的三个指针要保证区间为0
//最终的终止条件是i与two相同
if(len<2){
return;
}
int zeros=0;
int i=0;
int two=len;
while(i<two){
if(nums[i]==0){
swap(nums,i,zeros);
i++;
zeros++;
}else if(nums[i]==1){
i++;
}else{
two--;
swap(nums,i,two);
//此处不需要i++,two指针先前移,再交换,此时two指向的是一个新的小数,交换过后i指向的也是一个新的小数,需要继续判断,所以不需要移动i
}
}
}
private void swap(int[] nums,int index1,int index2){
int temp=nums[index1];
nums[index1]=nums[index2];
nums[index2]=temp;
}
}
笔记:
1、注意交换前后指针移动的情况。