1.快速排序
这个快排的代码是基于算法导论中的介绍写的,将数组中最后一个元素作为pivot,设置两个指针i,j,都从数组头开始,当j遇到比pivot小的元素时,就将i和j的元素交换,否则j一直向后搜寻,i保持不动,最后将i递增1,与pivot元素交换,此时pivot前面的都是比pivot小的元素,后面都是比pivot大的元素,再分别对前后两个区间进行重复的操作。直到完全排序。
#include <iostream>
#include <vector>
using namespace std;
int partition(vector<int> &nums, int start, int end) {
int i = start - 1;
for (int j = start; j < end; j++) {
if (nums[j] <= nums[end]) {
i = i + 1;
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
}
i = i + 1;
int temp = nums[i];
nums[i] = nums[end];
nums[end] = temp;
return i;
}
void quicksort(vector<int> &nums, int start, int end) {
if (start < end) {
int q = partition(nums, start, end);
quicksort(nums, start, q - 1);
quicksort(nums, q + 1, end);
}
}
int main() {
int a[7] = { 1,5,2,7,4,9,4 };
vector<int> nums(a,a+7);
int n = nums.size();
quicksort(nums, 0, n - 1);
for (int i = 0; i < n; i++) {
cout << nums[i] << '\t';
}
}
2.快速选择
这个快速选择代码是找到数组中第k个元素(k从0到n-1),思想和上面的快排一致,只不过现在找到了第k个元素所在的区间后,只需在所在区间进行下一次搜寻,另一个区间就不用管了。
#include <iostream>
#include <vector>
using namespace std;
int partition(vector<int> &nums, int start, int end) {
int i = start - 1;
for (int j = start; j < end; j++) {
if (nums[j] <= nums[end]) {
i = i + 1;
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
}
i = i + 1;
int temp = nums[i];
nums[i] = nums[end];
nums[end] = temp;
return i;
}
void quickselect(vector<int> &nums, int start, int end, int k) {
if (start == end||start==k) {
cout << nums[start];
return;
}
if (end == k) {
cout << nums[end];
return;
}
if (start < end) {
int q = partition(nums, start, end);
if (q == k) {
cout << nums[k];
return;
}
else if (q > k) {
quickselect(nums, start, q - 1, k);
}
else {
quickselect(nums, q + 1, end, k);
}
}
}
int main() {
int a[7] = { 1,5,2,7,4,9,4 };
vector<int> nums(a, a + 7);
int n = nums.size();
quickselect(nums, 0, n - 1,1);
}