给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶：
一个直观的解决方案是使用计数排序的两趟扫描算法。
首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
- 你能想出一个仅使用常数空间的一趟扫描算法吗？
代码
class Solution {
public void sortColors(int[] nums) {
if (nums.length == 1) return;
if (nums.length == 2) {
if (nums[1] == 0 || (nums[1] == 1 && nums[0] == 2)){
int temp = nums[0];
nums[0] = nums[1];
nums[1] = temp;
}
}else {
int v0 = 0,v1 = 0,v2 = 0;
for (int i = 0;i < nums.length;i++) {
if (nums[i] == 0){
v0++;
}else if (nums[i] == 1){
v1++;
}else {
v2++;
}
}
for (int i = 0;i < nums.length;i++) {
if (i < v0 && nums[i] != 0){
nums[i] = 0;
}else if (i >= v0 && i < (v0+v1) && nums[i] != 1){
nums[i] = 1;
}else if (i >= (v0+v1) && i < (v0+v1+v2) && nums[i] != 2){
nums[i] = 2;
}
}
}
}
}
- 官方解法，三指针
class Solution{
/*
荷兰三色旗问题解
*/
public void sortColors(int[] nums) {
int p0 = 0,curr = 0;
int p2 = nums.length - 1;
int tmp;
while (curr <= p2) {
if (nums[curr] == 0) {// 将0移动到头部
tmp = nums[p0];
nums[p0++] = nums[curr];
nums[curr++] = tmp;
} else if (nums[curr] == 2) {// 将2移动到尾部
tmp = nums[curr];
nums[curr] = nums[p2];
nums[p2--] = tmp;
}else {// 1不动
curr++;
}
}
}
}
转载请注明来源，欢迎对文章中的引用来源进行考证，欢迎指出任何有错误或不够清晰的表达。可以在下面评论区评论，也可以邮件至 1056615746@qq.com
