Data Structure_Sort Algorithm

  1. 云栖社区>
  2. 博客>
  3. 正文

Data Structure_Sort Algorithm

绿箭侠2017 2018-10-10 21:39:00 浏览716
展开阅读全文

排序算法

Tool implement

   //generate a array of n elements, range [rangL, rangeR]
    int *generateRandomArray(int n, int rangL, int rangeR) {
        assert(rangeR >= rangL);
        int *arr = new int[n];
        srand(time(NULL));
        for (int i = 0; i < n; i++) {
            arr[i] = rand() % (rangeR - rangL + 1) + rangL;
        }
        return arr;
    }

    template<typename T>
    void showArray(T arr[], int n) {
        for (int i = 0; i < n; i++) {
            cout << arr[i] << " ";
        }
        cout << endl;
    }

生成随机的n个数量的数组,输出数组每一个元素的内容。测试排序算法使用的标准就是运行时间和排序的正确性,所以需要一个验证正确性和计算排序时间的:

    template<typename T>
    void testSort( string sortName, void(*sort)(T[], int), T arr[], int n){
        clock_t startTime = clock();
        sort(arr, n);
        clock_t endTime = clock();
        if (isSorted(arr, n)){
            cout << sortName << ": " << double(endTime - startTime)/CLOCKS_PER_SEC << "s" << endl;
        } else
            cout << "sort function is wrong!" << endl;
        return;
    }

    template<typename T>
    bool isSorted(T arr[], int n){
        for (int i = 0; i < n-1; i++){
            if (arr[i] > arr[i+1])
                return false;
        }
        return true;
    }

O(n^2)的排序

这种算法应该是属于比较慢的算法了,编码简单,易于实现,是一些简单场景的选择。在一些特殊情况下,简单的排序算法更容易实现且更有效。从简单的算法演变到复杂的算法,循序渐进,作为子过程改进跟复杂的算法。

选择排序——SelectionSort

选择排序比较简单,如果是最简单的一个了。基本思想就是遍历每一个位置,选择这个位置之后最小的元素与当前元素对换:

img_586c78cb6f42bb9e429788dc3813c492.png
比如这个数组,一开始是第一个位置,然后选择2到n个位置最小的元素,找到就是1了,然后和第一个元素互换:
img_d3f058601b2bbe09236d85c92fb17c3a.png
也就是挨个挨个位置找到最小的元素。很明显,毫无疑问,他的复杂度肯定就是
O(n^2)

代码实现

template<typename T>
void SelectionSort(T arr[], int n) {
    for (int i = 0; i < n; i++) {
        int minIndex = i;
        for (int j = i; j < n; j++) {
            if (arr[j] < arr[minIndex])
                minIndex = j;
        }
        swap(arr[minIndex], arr[i]);
    }
}

由于是模板类,所以还可以用来排序string,运算符重载还可以用来对类排序。


img_e36105e23188361c29bcddeb2a625a9c.png

img_b77f5faf0fedee02389014d404195717.png

定义一个学生类用于排序:

    struct student{
        string name;
        float score;

        bool operator<(const student &otherStudent){
            return score < otherStudent.score;
        }

        friend ostream& operator<<(ostream &os, const student &otherStudent){
            os << "Student: " << otherStudent.name << " " << otherStudent.score << endl;
            return os;
        }
    };

img_6fd3362f17ec3981008422a1ccf5792a.png

img_3319cd4e5cc2c8575a22e0e040fd30f6.png

插入排序——InsertionSort

插入排序也是一个O(n^2)的排序算法。

img_250b3f1201ee634615fbc3f27dc98dc4.png
一开始先比较了第一和第二位,如果顺序是正确的,那么就不换,继续比较第三第四位,2比8小交换位置,比6有小交换位置,直到适合它的位置即可。
img_9f006c207ff6596feb13271b49c0cbfb.png
以此类推最后排序完成。如果要判断的元素是刚刚好大于之前的元素的时候,那么就不用比较了,所以插入排序操作是可以比选择排序要快的,来看一些代码实现:

template<typename T>
void InsertionSort(T arr[], int n){
    for (int i = 1; i < n; i ++){
        for (int j = i; j > 0; j --){
            if (arr[j] < arr[j-1]){
                swap(arr[j], arr[j-1]);
            } else
                break;
        }
    }
}

img_7ce00d13093515a46119ab3eac38106f.png

img_52321317ed7dd2b7ffbad2ca07c1ee19.png
然而事与愿违,时间反而多了一倍。这是和插入排序的一些交换操作引起的,插入排序虽然可以终止某些交换操作,但是在遍历的时候同时也做交换了,每一次交换就有三次赋值的操作,再加上索引数组这些时间,就远大于选择排序了。为了减少这种交换的次数,我们改进一下:
img_57fb641d06815993c74abcc8c84a72a2.png
一开始只做比较不做交换,复制一份6出来,比较6和8,如果8是大于6的,就证明这个8是不应该在当前位置的,所以8覆盖6,然后6往前移动,按照正常套路进行比较下去。
代码实现:

template<typename T>
void InsertionSort_version2(T arr[], int n){
    for (int i = 1; i < n; i ++){
        T temp = arr[i];
        int j = 0;
        for (j = i; j > 0 && arr[j-1] > temp; j --){
            arr[j] = arr[j-1];
        }
        arr[j] = temp;
    }

这次就正常了:
img_aaf5cca76390858f7efaac987a6e5cf4.png

对于插入排序还有一个比较好的性质,如果当前数组是一个基本有序的数组那么基本插入排序每一次就基本被终止了,那么就会很快。

O(nlogn)的排序

在小数据量的情况下,两者的差别都不会特别大,但是当数据量变大的时候,两者的差别就会越来越大了。

归并排序——MergeSort

img_6ec59b0f418f91a5861e3a784c97b6eb.png

步骤其实很简单,首先进行划分,当划分到一定的细度的时候就不能划分了,然后就进行归并
img_a464ed72763a4de0aab984d980fcf954.png
然后往上归并即可。划分的时间是
logN
,归并的时候就是
N
了,所以一共就是
NlogN
级别了。那么主要的困难就是归并的时候了:
img_cf4bf608be1674911dfe98f6d6d2fee9.png
如何把如上的两个归并的数组合并?
img_09a7ab34a5a2641ab16d374924b3e1b2.png
需要用到三个索引,两个红色的索引代表的就是已经排好序的两个数组,蓝色的指向了一个临时空间:
img_f10b0b1d1ed91c0b60a08aa324b2eb6d.png
一开始2和1比较,1比较小,那么1就填在了蓝色索引的位置,蓝色索引加一,因为1填进去了,所以右边的红色索引加一,以此类推,到最后哪边没有完就直接整块扔进去就好了。代码实现:

template<typename T>
void __merge(T arr[], int l, int mid, int r) {
    T *temp = new T[r-l+1];
    for (int i = l; i <= r; i++) {
        temp[i - l] = arr[i];
    }
    int i = l, j = mid + 1;
    for (int k = l; k <= r; k++) {
        if (i > mid) {
            arr[k] = temp[j - l];
            j++;
        } else if (j > r) {
            arr[k] = temp[i - l];
            i++;
        } else if (temp[i - l] < temp[j - l]) {
            arr[k] = temp[i - l];
            i++;
        } else {
            arr[k] = temp[j - l];
            j++;
        }
    }
    delete[] temp;
}

template<typename T>
void __mergeSort(T arr, int l, int r) {
    if (l >= r) {
        return;
    } else {
        int mid = (l + r) / 2;
        __mergeSort(arr, l, mid);
        __mergeSort(arr, mid + 1, r);
        __merge(arr, l, mid, r);
    }
}

template<typename T>
void MergeSort(T arr[], int n) {
    __mergeSort(arr, 0, n - 1);
}

最主要的就是Merge函数了。对于temp数组是存在有偏移的,所以是需要减去一个偏移量,因为到最后l和r不一定是从0开始的。
img_c3ccd82a14bf3d54ef3c31f9c55bf591.png

可以看到对于一万个数组排序的时间很快,其实归并排序就是典型的用空间换时间的,每一次递归就开了一个空间,每一次递归的时候还有一个缓冲数组,所以使用的空间复杂度比一般的排序算法要大。

如果是十万个数据那么差距就更加大了:
img_c7e2af56362edb8b0835f0082184ecd4.png

但是对于这个代码来说,还是可以优化的,因为在归并的时候,我们是无论两边的数组是个什么情况都进行一个归并,但是事实上这样是不严谨的,因为事实上如果一边的最大的元素小于另一边的最小的元素,就可以说完全大于了,所以加上判断:
void __mergeSort(T arr, int l, int r) {
    if (l >= r) {
        return;
    } else {
        int mid = (l + r) / 2;
        __mergeSort(arr, l, mid);
        __mergeSort(arr, mid + 1, r);
        if (arr[mid] > arr[mid + 1]) {
            __merge(arr, l, mid, r);
        }
    }
}

加上一个判断即可。

自底向上的归并排序——upMergeSort

其实思想都是一样的,都是分而治之的思想,先两个两个的排,然后四个四个的排,然后八个八个的排,以此类推:
img_a74abd926f5667ef9af446159fcb8755.png

先两个两个的排,然后四个四个:
img_21c6b65752b7e0ba13266a33ccb9deab.png

之后8个一起,这个过程其实不需要递归,直接迭代实现即可。
template<typename T>
void upMergeSort(T arr[], int n){
    for (int sz = 1; sz <= n; sz = sz + sz){
        for (int i = 0; i + sz < n; i += sz + sz){
            __merge(arr, i, i+sz-1, min(i+sz+sz-1, n-1));
        }
    }
}

可以看到代码量很少。
img_aa74abc80385ff5415e436f7b90e554b.png

时间差不多。

快速排序——QuickSort

归并排序和快速排序不同的就是,归并排序是无论什么情况直接就一分为二两部分,快速排序也是分成两部分,但是快速排序是先找到对应的一个元素的位置,然后直接找到这个元素应该在的位置:

img_9253f5f12f30c1e4787de3678532ed18.png
比如4,找到这个元素所在的位置。这个时候这个数组就有一个性质了,左边的元素一定比4小,右边的一定比4要大。那么接下来的主要操作就是在于如何找到4这个元素对应的位置。
img_73f7be40b0f7d0aae1b7a5160fe91842.png
v是要分割的数字,e就是当前要判断的数字。有两个情况要判断,如果是大于V的,那就简单了,直接i++,如果是小于V,那就把i和j+1的数字进行交换即可,然后j++,最后i++即可。
img_d7c15a8362b487019ab89be292bc87c0.png
最后就是这个样子,只需要j和l换即可。
代码实现:

template<typename T>
int __partition(T arr[], int l, int r) {
    T v = arr[l];
    int j = l;
    for (int i = l + 1; i <= r; i++) {
        if (arr[i] < v) {
            swap(arr[j + 1], arr[i]);
            j++;
        }
    }
    swap(arr[j], arr[l]);
    return j;
}

template<typename T>
void __quickSort(T arr[], int l, int r) {
    if (l >= r) {
        return;
    }
    int p = __partition(arr, l, r);
    __quickSort(arr, l, p - 1);
    __quickSort(arr, p + 1, r);
}

template<typename T>
void QuickSort(T arr[], int n) {
    __quickSort(arr, 0, n - 1);
}

img_5791237db181473c520a1f9c4ff532c1.png
.可以看到效果还是有的。快速排序之所以是
nlogn
是因为每一次都会分成两个,但是是不均等的两个:
img_77caba7908dfeb571489bf1744487a77.png
这样,问题就来了,如果这个数组是一个基本有序的数组,那么大概就是这样了:
img_d2820dcdbca94e69630bb9303dc766ea.png
退化成了
O(n^2)
的问题,因为我们选择的是第一个个元素进行排序,所以改进我们就可以随机选择一个元素做分割,改进一下。选择一个随机:

template<typename T>
int __partition(T arr[], int l, int r) {
    swap(arr[l], arr[rand()%(r-l+1)+l]);
    T v = arr[l];
    int j = l;
    for (int i = l + 1; i <= r; i++) {
        if (arr[i] < v) {
            swap(arr[j + 1], arr[i]);
            j++;
        }
    }

效果就不出了。
接下来还有一个问题,如果这个数组是一个基本有序的数组,测试一下数据:

img_6e30e4806063a9f3520ea73f7eda8696.png

可以看到快速排序还是慢了点。为什么会这样,我们考虑一下partition函数:
img_8af81630e9c18bbcd619e3424b26560b.png
当拥有大量重复元素的时候问题来了:
img_2ac5d71fae3c8374a2ec375f216d3f87.png
那么切分的时候自然又回到刚刚的情况了。
img_3e8599222824d27dbf3cdd8f345bf762.png
我们可以用这种方法进行改进,把大于和小于的放在两边,找到i和j大于V和小于V就停下,交换两个的值。
img_79936c95e8b84a158dbe273123385a87.png

按照如上的这一种结构就算是有大量的重复的数值也可以平均分开,达到两边是相同的数量。
代码实现:

template<typename T>
int __partition_version2(T arr[], int l, int r) {
    swap(arr[l], arr[rand() % (r - l + 1) + l]);
    T v = arr[l];
    int i = l+1, j = r;
    while (true){
        while (i <= r && arr[i] < v){
            i++;
        }
        while (j >= l+1 && arr[j] > v){
            j--;
        }
        if (i > j){
            break;
        } else{
            swap(arr[i], arr[j]);
            i++;
            j--;
        }
    }
    swap(arr[l], arr[j]);
    return j;
}

img_7ab5e9615de68ea46230e297c5b32d80.png

这样效果就好很多了。
实现带有重复元素的数组快速排序还有一个更加经典的实现方法,就是三路排序了。
img_df0d4fcbeaf23777f64af673763a69db.png

代码实现:


template<typename T>
int __partition_version3(T arr[], int l, int r){
    swap(arr[l], arr[rand() % (r - l + 1) + l]);
    T v = arr[l];
    int lt = l, gt = r+1, i = l+1;
    while (i < gt){
        if (arr[i] < v){
            swap(arr[i], arr[lt+1]);
            lt++;
            i++;
        } else if (arr[i] > v){
            swap(arr[i], arr[gt-1]);
            gt--;
        } else{
            i++;
        }
    }
    swap(arr[l], arr[lt]);
    return lt;
}

堆排序——HeapSort

堆是一种比较常用的数据结构,常用的堆有二叉堆,费波纳茨堆等等。堆是一颗完全二叉树,除了最后一层其他的每一层都是满的,而最后一层都的子节点都是集中在了左边。堆的实现在DataStructure里面提到,实现很简单。因为实现的是最大堆,出来的元素就是最大的:

using namespace std;
template<typename T>
void HeapSort_version1(T arr[], int n){
    MaxHeap<T> heap = MaxHeap<T>(n);
    for (int i = 0; i < n; ++i) {
        heap.insert(arr[i]);
    }
    for (int j = n-1; j >= 0; --j) {
        arr[j] = heap.pop();
    }
}
img_d112d240df53e0cfa2df9b022a248990.png

但是有一个小问题,每一次都是插入这样效率不高,所以我们可以直接在原数组上操作


img_7f147f5782caeb407594ce942dabee0f.png

比如这就并不是一个正确的最大堆,需要用heapify转变一下。注意到每一个叶子都是一个堆,而且是符合规定的堆,所以肯定是从最底层开始。第一个不是堆的节点一定是最后一个叶子除2。所以第一个要做的就是对第一个不是堆的节点进行排序,很明显就是22这个元素了,对比交换,这个过程其实就是一个shitDown的过程,而我们要shitdown的元素是那5个不是堆的元素,所以对于这个数组直接做5次循环对每一个不是堆的元素进行shiftDown操作即可。

    MaxHeap(item arr[], int n){
        data = new item[n+1];
        capacity = n;
        for (int i = 0; i < n; ++i) {
            data[i+1] = arr[i];
        }
        count = n;
        for (int j = count/2; j >= 1; --j) {
            shiftDown(j);
        }
    }

之后就可以使用heapify来代替迭代插入,其实效果差别不大。堆排序在现实中其实蛮少用到的,常用的还是对于动态数据结构的维护。

img_fa0e594c2e294152412af853b52bbad6.png

事实上这个版本的和上一个版本的差别是不大的。刚刚的一个一个元素插入堆里面的计算复杂度是
O(nlogn)
,而使用heapify的复杂度是
O(n)
。上面的排序每一次都是复制一个数组出来排序,但是如果原地在原数组上进行排序,速度是可以快很多的,这个就是原地堆排序。
img_cbca9f0da6598eca1cf000ea1608e8ac.png
当前数组是一个最大堆,那么他的第一个元素就是v,那么v就是最大的元素,这个最大的元素应该在最后,所以和最后一个元素调换位置;
img_9951e19b97ad885fc944e209f9b49745.png
这个时候前面的几个元素就不是堆了。这个时候继续对第一个元素进行shifDown操作,之后再交换即可。但是如果是这样,那就需要调整一下索引,之前是1开始,那么现在就要从0开始了。
img_ef7ac05ec119f63e31cde67f09e18d5d.png
自然公式就要变一下。这个时候最后一个非叶子节点的索引就是(count-1)/2。所以,第一步就是先建立一个堆,就在当前的数组上建立;建立好堆之后那么第一个元素就是最大的元素了,这个时候最大的元素应该是在最后的,所以第二部就是要将第一个元素和最后一个元素交换,然后把最后一个元素之前的(n-1)个元素进行heapify操作,之后又交换又heapify直到前面就剩下一个元素即可。所以首先要重写shiftDown操作,因为这个操作的索引已经是从0开始了:

template<typename T>
void __shiftDown(T arr[], int n, int i){
    while (2*i + 1 < n){
        int change = 2*i + 1;
        if (change + 1 < n && arr[change] < arr[change+1]){
            change ++;
        }
        if (arr[i] >= arr[change]){
            break;
        }
        swap(arr[i], arr[change]);
        i = change;
    }
}

接下了就是按照上述流程排序:

template<typename T>
void HeapSort_version2(T arr[], int n) {
    //heapify,create a max heap;
    for (int i = (n-1)/2; i >= 0; --i) {
        __shiftDown(arr, n, i);
    }
    for (int j = n-1; j > 0 ; --j) {
        swap(arr[0], arr[j]);
        __shiftDown(arr, j, 0);
    }
}

看看效果:


img_b5bd7c86522d8a6760447ed6f7a1cca1.png

差别不是特别大,但是总的来说还是快了,因为它不需要进行额外的空间,也不需要赋值的操作等等,相对会快点。

排序算法总结

平均时间复杂度 原地排序 额外空间 稳定性
Insertion Sort O(n^2) O(1)
Merge Sort O(nlogn) O(n)
Quick Sort O(nlogn) O(logn)
Heap Sort O(nlogn) O(1)

快速排序的额外空间是logn,因为递归的次数就是logn,而每一次递归就需要一个栈空间来存储。

最后附上GitHub代码:https://github.com/GreenArrow2017/DataStructure/tree/master/SortAlgorithm

网友评论

登录后评论
0/500
评论
绿箭侠2017
+ 关注