颜色分类
解法一:遍历数组
public class LeetCode_075 {
public static void sortColors(int[] nums) {
if (nums.length == 1) {
return;
}
// zeroIndex用来记录从左开始第一个不为红色的位置,twoIndex用来记录从右开始第一个不为蓝色的位置
int zeroIndex = 0, twoIndex = nums.length - 1;
for (int i = 0; i <= twoIndex; ) {
if (nums[i] == 2) {
// 如果当前的值等于2,则将当前位置的值和twoIndex位置的值进行交换
nums[i] = nums[twoIndex];
nums[twoIndex] = 2;
twoIndex--;
continue;
} else if (nums[i] == 1) {
i++;
} else if (nums[i] == 0) {
// 如果当前的值等于0,则将当前位置的值和zeroIndex位置的值进行交换
nums[i] = nums[zeroIndex];
nums[zeroIndex] = 0;
zeroIndex++;
i++;
}
}
}
public static void main(String[] args) {
int[] nums = new int[]{2, 0, 2, 1, 1, 0};
sortColors(nums);
for (int num : nums) {
System.out.print(num + " ");
}
}
}