【基础算法】（1）快速排序

快速排序——算法思路

本栏目为学习ACWINGy神的算法基础课的笔记

1. 暴力思路（消耗空间大）
   需要多开两个数组分别存储小于x与大于x的数
   
2. 优美思路（双指针）

算法代码

#include <iostream>
using namespace std;
const int N = 100010;
int q[N];
void quick_sort(int q[], int l, int r)
{
    //l为左边界，r为右边界
    if(l>=r) return;//如果只有一个数即不用排序
    //第一步：确定分界点 
    //注意：x一定要是上取整不然会出现边界问题
    int i = l - 1, j = r + 1, x = q[l + r >> 1];
    //第二步：左指针指向大于等于x的地方停住，右指针指向小于等于x的地方停下，两者交换继续
    while(i < j)
    {
        do i++;while(q[i] < x);
        do j--;while(q[j] > x);
        if(i < j) swap(q[i], q[j]);
    }
    //第三步：递归处理左右两边
    quick_sort(q,l,j);
    quick_sort(q,j+1,r);
}

int main()
{
    int n;
    scanf("%d", &n);
    for (int i = 0; i < n; i ++ ) scanf("%d", &q[i]);
    quick_sort(q, 0, n - 1);
    for (int i = 0; i < n; i ++ ) printf("%d ",q[i]);
    return 0;
}

习题练习

第k个数

给定一个长度为 n 的整数数列，以及一个整数 k，请用快速选择算法求出数列从小到大排序后的第 k 个数。
一定要独立多打几遍代码

习题代码

#include <iostream>

using namespace std;
const int N = 100010;
int q[N];

void quick_sort(int q[], int l, int r)
{
    if(l>=r) return;
    int i=l-1, j=r+1, x=q[l+r >> 1];
    while(i<j)
    {
        do i++;while(q[i]<x);
        do j--;while(q[j]>x);
        if(i<j) swap(q[i],q[j]);
    }
    quick_sort(q,l,j);
    quick_sort(q,j+1,r);
}

int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=0; i<n; i++) scanf("%d",&q[i]);
    quick_sort(q,0,n-1);
    printf("%d",q[k-1]);
    return 0;
}

算法注意点

1. 注意数组边界问题，有不同偏好的写法但是要注意越界情况。比如x若取l+r/2的值的话，一定要上取整，写成x=(l+r+1)/2或者x=l+r >> 1
2. 由于使用do-while循环,所以i和j一定自增,使得循环会继续下去。但是如果采用while循环(i和j的初始化做出对应的变更),i和j在特殊情况下不自增的话,循环就会卡死。