给定一个包含红,白,蓝且长度为n的数组,将数组元素进行分类使相同颜色的元素相邻,并按照红、白、蓝的顺序进行排序。
我们可以使用整数0,1和2分别代表红,白,蓝。
您在真实的面试中是否遇到过这个题?
Yes
样例
注意
不能使用代码库中的排序函数来解决这个问题
说明
一个相当直接的解决方案是使用计数排序扫描2遍的算法。
首先,迭代数组计算0,1,2出现的次数,然后依次用0,1,2出现的次数去覆盖数组。
你否能想出一个仅使用常数级额外空间复杂度且只扫描遍历一遍数组的算法?
解法:常数级额外空间复杂度且只扫描遍历一遍数组的算法,使用双指针
class Solution {
/**
* @param nums: A list of integer which is 0, 1 or 2
* @return: nothing
*/
public void sortColors(int[] nums) {
int redTail = 0;
int blueHead = nums.length - 1;
int cur = 0;
while(cur <= blueHead) {
if(nums[cur] == 0) {
swap(nums, redTail, cur);
redTail++;
cur++;
}
else if(nums[cur] == 2) {
swap(nums, blueHead, cur);
blueHead--;
}
else {
cur++;
}
}
}
private void swap(int[] nums, int b, int e) {
int tmp = nums[b];
nums[b] = nums[e];
nums[e] = tmp;
}
}