主要参考:
第 6 节 三路快排
分治算法
分而治之
条件:
(1)原问题可以分解成若干个规模比较小的相同子问题。
(2)子问题相互独立。
(3)子问题的解可以合并为原问题的解。
快速排序
划分函数 partition
写法一:
int partition(vector<int>& arr, int low, int high){
int i=low,j=high;
int base=arr[low];//选取第一个元素作为基准
while(i<j){
while(i<j && arr[j]> base ){//high从右往左扫描 寻找小于base的值
j--;
}
if(i<j){//交换
swap(arr[i], arr[j]);
i++;
}
while(i<j && arr[i]<=base){//high从左往右扫描 寻找大于base的值
i++;
}
if(i<j){
swap(arr[i], arr[j]);
j--;
}
}
return i;
}
写法二:
int partition(vector<int>& arr, int low, int high) {
if(low > high)
return low;
int i=low;
int j=high;
int base=arr[low];
while(i<j){
while(i<j && arr[j]>=base)
j--;
while(i<j && arr[i]<=base)
i++;
if(i<j){
int t = arr[i];
arr[i] = arr[j];
arr[j] = t;
//swap(arr[i],arr[j]);
}
}
arr[low]=arr[i];
arr[i]=base;
//swap(arr[low],arr[i]);
return i;
}
快排递归算法 quick_sort
void quick_sort(vector<int>& arr, int start, int end){
int pos;
if (start<end){
pos = partition(arr, start, end);
quick_sort(arr,start,pos-1);
quick_sort(arr,pos+1,end);
}
return;
}
三划分快排
详细可看视频:第 6 节 三路快排
void Quick_Sort(vector<int> &nums, int left, int right){
if (left >= right)
return;
int lt = left+1;
int gt = right;
int i = left+1;
swap(nums[left], nums[rand() % (right - left + 1) + left]);
int temp = nums[left];
while(i <= gt){
if(nums[i] < temp){
swap(nums[i], nums[lt]);
i++;
lt++;
}
else if(nums[i] == temp){
i++;
}
else if(nums[i] > temp){
swap(nums[i], nums[gt]);
gt--;
}
}
swap(nums[left], nums[lt-1]);
Quick_Sort(nums, left, lt-1);
Quick_Sort(nums, gt+1, right);
}
leetcode部分题目
75. 颜色分类
75. 颜色分类
快排:
class Solution {
public:
int partition(vector<int>& arr, int low, int high){
int i=low,j=high;
int base=arr[low];//选取第一个元素作为基准
while(i<j){
while(i <j && arr[j]> base ){//high从右往左扫描 寻找小于base的值
j--;
}
if(i<j){//交换
swap(arr[i], arr[j]);
i++;
}
while(i<j && arr[i]<=base){//high从左往右扫描 寻找大于base的值
i++;
}
if(i<j){
swap(arr[i], arr[j]);
j--;
}
}
return i;
}
void quick_sort(vector<int>& arr, int start, int end){
int pos;
if (start<end){
pos = partition(arr, start, end);
quick_sort(arr,start,pos-1);
quick_sort(arr,pos+1,end);
}
return;
}
void sortColors(vector<int>& nums) {
//快排 以1为基准 扫描一次可以得到最终结果
quick_sort(nums,0,nums.size()-1);
}
};