Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue.
Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.
Note:
You are not suppose to use the library’s sort function for this problem.
排序O(Nlog(N))这道题是不行的。。
0 0 2 1 1 0 2 0 1 0 2
想了一下不知道是不是可以这样做:
左边都得是0 右边都得是2 1的位置是根据02个数确定的
一个左指针 遇到0就右走 遇到12的停住
一个右指针 遇到2就左走 遇到01就停住
那么此时可能是:
这样4种情况:
前3种比较好处理:
1.两边调换之后 左指针走一步
2.两边调换之后 左右指针都走一步
3.两边调换之后 右指针走一步
4.这时候要考虑 中间是不是都是1
如果从左到右遍历一遍都是1 那么就结束
如果发现了不是1的数(0或2)那么:
发现的是0:和左指针对换 左指针走一步
发现的是1:和右指针对换 右指针走一步
public void sortColors(int[] nums) {
for(int i = 0,j = nums.length-1;i<j;){
if(nums[i] == 0) i++;
if(nums[j] == 2) j--;
if(i >= j) break;
if(nums[i] == 2 && nums[j] == 0){
swap(nums, i, j);
i++;
j--;
}else if(nums[i] == 1 && nums[j] == 0){
swap(nums, i, j);
i++;
}else if(nums[i] == 2 && nums[j] == 1){
swap(nums, i, j);
j--;
}else{
int k = i+1;
for(;k < j;k++){
if(nums[k] == 0){
swap(nums, i, k);
i++;
break;
}
else if(nums[k] == 2){
swap(nums, k, j);
j--;
break;
}
}
if(k == j)
break;
}
}
}
public void swap(int[] nums,int i,int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}