荷兰国旗问题——LeetCode75
荷兰国旗问题描述如下:
给定一个整数数组和一个值M(存在于原数组中),要求把数组中小于M的元素放到数组的左边,等于M的元素放到数组的中间,大于M的元素放到数组的右边,最终返回一个整数数组,只有两个值,0位置是等于M的数组部分的左下标值、1位置是等于M的数组部分的右下标值。
LeetCode75描述如下:
给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用库内置的 sort 函数的情况下解决这个问题。
方法一:
使用辅助空间,分别记录数组中0、1以及2元素的个数
void sortColors(vector<int>& nums) {
int colors[3]; //存储各个元素的数量
for(int i = 0;i<3;i++){
colors[i] = 0;
}
for(int i = 0;i<nums.size();i++){
if(nums[i] == 0){
colors[0]++;
}
else if(nums[i] == 1){
colors[1]++;
}
else{
colors[2]++;
}
}
int xia = 0; //指示颜色中的下标
for(int k = 0;k<nums.size();k++){
while(colors[xia]==0){
xia++;
}
nums[k] = xia;
colors[xia]--;
}
}
方法二:
遍历第一次,将0元素全部放在数组开头(使用交换);遍历第二次,将元素1接着放置(或者将元素2放置在数组末尾,一样的道理);那么剩下的2全部都在末尾(全部的1在中间)
方法三:
使用双指针优化方法二。p0指向下一个0应该存放的位置,p1指向下一个1应该存放的位置,还有一个指针用来遍历。遍历规则如下:
若遍历到1,将其与p1处元素交换;
若遍历到0,将其与p0处元素交换。但是此时可能导致原本的1被换到2“堆”中。所以还需要一步,将其再与p1处元素交换。
(待完成,补充图)