要求:
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
思路:
选用三个指针,p0,cur,p1。p0代表0的最右边界,p2代表2的最左边界,cur也就主要用于判断的移动指针。只要cur<=p2就进入循环。
-
初始化:p0=cur=0;p2=nums.length-1;
-
判断条件:
1. nums[cur] = 0,交换nums[cur]和nums[p0],cur++,p0++ 2. nums[cur]=2,交换nums[cur]和nums[p2],cur不变,p2-- 3. nums[cur]=1,cur++
代码:
class Solution {
public void sortColors(int[] nums) {
int cur = 0;
int p0 = 0;
int p2 = nums.length-1;
while (cur <=p2) {
if(nums[cur] == 2) {
int temp = nums[cur];
nums[cur] = nums[p2];
nums[p2--] = temp;
} else if(nums[cur]==1) {
cur++;
} else {
int temp = nums[cur];
nums[cur] = nums[p0];
nums[p0] = temp;
p0++;
cur++;
}
}
}
}