三路快排是快速排序算法的升级版,用来处理有大量重复数据的数组。
主要思想是选取一个key,小于key的丢到左边,大于key的丢到右边,递归实现即可。
具体操作过程参考:https://blog.csdn.net/k_koris/article/details/80585979
C++代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
#include<iostream> using namespace std; void swap(int &a, int &b){ int t = a; a = b; b = t; } void triSort(int *a, int low, int high){ if(low >= high)return; int key = a[low]; int i = low, j = low; int k = high; while( i <= k){ if(a[i] < key)swap(a[i++], a[j++]); else if(a[i] > key)swap(a[i], a[k--]); else i++; } triSort(a, low, j); triSort(a, k + 1, high); } int main(){ int n[12] = {5, 9, 0, 1, 6, 3, 8, 7, 2, 4, 4, 4}; triSort(n, 0, 11); for(int i = 0; i < 12; i++){ cout<<n[i]<<" "; } cout<<endl; return 0; } |
效果:
LeetCode75 颜色分类
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
1 2 |
<strong>输入:</strong> [2,0,2,1,1,0] <strong>输出:</strong> [0,0,1,1,2,2] |
进阶:
- 一个直观的解决方案是使用计数排序的两趟扫描算法。
首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。 - 你能想出一个仅使用常数空间的一趟扫描算法吗?
代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
class Solution { public: void swapx(int &a, int &b){ int t = a; a = b; b = t; } void sortColors(vector<int>& nums) { int i = 0, j = 0, k = nums.size()-1; int key = 1; while(i <= k){ if(nums[i] < key)swapx(nums[i++], nums[j++]); else if(nums[i] > key)swapx(nums[i], nums[k--]); else i++; } } }; |