逖靖寒的世界

每天进步一点点

导航

快速排序之美

快速排序的实现是优美的,但是很不幸,我之前的实现非常差劲。参见《分治法:用C#实现快速排序

今天,我用C++重写了一遍,精简代码,力求优美。

#include <iostream>
 
using namespace std;
 
int partition1(int* a, int l, int r)
{
    int m = l;
    for (int i = l; i < r - 1; i++) {
        if (a[i] < a[r]) {
            int temp = a[i];
            a[i] = a[m];
            a[m] = temp;
            m++;
        }
    }
    int temp = a[r];
    a[r] = a[m];
    a[m] = temp;
 
    return m;
}
 
void quickSort(int* a, int l, int r)
{
    if (l <= r) {
        int q = partition1(a, l, r);
        quickSort(a, l, q - 1);
        quickSort(a, q + 1, r);
    }
}
 
int main() {
    int a[] = {4,66, 54, 3, 54,21, 31, 100, 9};
 
    quickSort(a, 0, 8);
 
    return 0;
}

 

接下来,在快速排序的基础上,进行少量的修改,让其在O(n)时间内,求出数组的前k个最小值。

只需要将上面的quickSort修改为:

 

void sortMinimum(int* a, int k, int l, int r)
{
    if (l <= r) {
        int q = partition1(a, l, r);
        if (q < k) {
            sortMinimum(a, k, q + 1, r);
        }
    }
}
即可。

posted on 2008-10-19 13:22  逖靖寒  阅读(762)  评论(7编辑  收藏  举报