力扣HOT100之技巧:75. 颜色分类
这道题实际上就是让我们不用sort()
函数来实现对原数组的排序,这里我直接使用快速排序对原数组进行排序了,也是复习一下基于快慢指针的快速排序写法。面试手撕快排的思路参考这个视频。
用时击败100%,还行。下面直接贴代码。
class Solution {
public:void sortColors(vector<int>& nums) {quicksort(nums, 0, nums.size() - 1);}void quicksort(vector<int>& nums, int begin, int end){if(begin >= end) return ; //只有一个元素,无需排序,直接返回//排序主逻辑int pivot = nums[end];int slow = begin, fast = begin; while(fast < end){if(nums[fast] > pivot)fast++;else{swap(nums[slow], nums[fast]);slow++;fast++;}}swap(nums[slow], nums[end]); //将枢轴移动到正确位置quicksort(nums, begin, slow - 1); //对枢轴左侧排序quicksort(nums, slow + 1, end); //对枢轴右侧排序}
};
但是,使用快速排序无法通过一次扫描就将整个数组排序完成,还可以使用更高效的方法。我感觉灵神的题解也写得挺通俗易懂的,感觉灵神的题解写的特别好,还是建议去看他的题解
以下是根据灵神的题解写出来的代码。
class Solution {
public:void sortColors(vector<int>& nums) {int p0 = 0, p1 = 0;for(int i = 0; i < nums.size(); i++){int temp = nums[i];nums[i] = 2; if(temp <= 1){nums[p1] = 1;p1++;}if(temp == 0){nums[p0] = 0;p0++;}}}
};