给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶:
- 一个直观的解决方案是使用计数排序的两趟扫描算法。
- 首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。
- 你能想出一个仅使用常数空间的一趟扫描算法吗?
这个题提示的思路就是直接两趟扫描计数,这个比较简单,使用C++编写,另外一种可以利用只有三种颜色,红色开头,蓝色结尾,那么只需要设置两个指针,红指针指向头和蓝指针指向尾,如果碰到0就与红指针指向的数字交换,然后红指针加1,如果碰到2,那么与蓝色指针指向的数字交换,然后蓝色指针减1,需要注意的是,由于遍历都是从前往后,所以防止出现将0换到前面的情况,与蓝指针交换完之后需要退一步,否则就会出现跳过遍历的情况,另外,遍历时,直接遍历蓝色指针个数即可,这个数字是不停变换的,当当前步大于蓝色指针时,说明已经是排好序了。
C++源代码:
class Solution {
public:
void sortColors(vector<int>& nums) {
int n = nums.size();
int count[3] = {0};
int k = 0;
for (int i=0;i<n;i++)
count[nums[i]]++;
for (int i=0;i<3;i++)
for(int j=0;j<count[i];j++)
nums[k++] = i;
}
};
python3源代码:
class Solution:
def sortColors(self, nums):
"""
:type nums: List[int]
:rtype: void Do not return anything, modify nums in-place instead.
"""
n = len(nums)
r = 0
b = n-1
i = 0
while i <= b:
if nums[i] == 0:
nums[r], nums[i] = nums[i], nums[r]
r = r + 1
elif nums[i] == 2:
nums[b], nums[i] = nums[i], nums[b]
b = b - 1
i = i - 1
i = i + 1