给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
一次循环,设置 2 个索引分别记录 red =0和blue=length的位子,1个索引i记录遍历的位子。
遇到0则放到red位子,需将red内容与i交换.i++
遇到2则放到blue位子,需将blue内容与i交换。
遇到1则不做改变 i++.
一遍过后,将red全部放在了前面,blue全部放在了后面.
class Solution {
public void sortColors(int[] nums) {
if(nums.length == 0 || nums.length == 1){
return;
}
int r = -1;
int i = 0;
int b = nums.length;
while(i < b){
if(nums[i] == 0){
r++;
int temp = nums[r];
nums[r] = nums[i];
nums[i] = temp;
i++;
}else if(nums[i] == 2){
b --;
int temp = nums[b];
nums[b] = nums[i];
nums[i] = temp;
}else{
i++;
}
}
}
}