資料結構與演算法——歸併排序

readyao發表於2016-01-11
歸併排序以O(NlogN)最壞情形執行時間執行;這個演算法是經典的分治策略,它將問題分成一些小的問題然後遞迴求解,而治的階段則是將分的階段解得的各答案修補在一起。分治是遞迴非常有力的方法。

基本思路:


原始碼:

/*************************************************************************
	> File Name: merge_sort.cpp
	> Author: 
	> Mail: 
	> Created Time: 2016年01月11日 星期一 15時16分26秒
 ************************************************************************/

#include <iostream>
#include <vector>
#include <algorithm>
#include <iterator>
using namespace std;

template<class T>
void merge(vector<T> & a, vector<T> & tmpArray, int leftPos, int rightPos, int rightEnd)
{
    int leftEnd = rightPos - 1;
    int tmpPos = leftPos;
    int numElements = rightEnd - leftPos + 1;

    while(leftPos <= leftEnd && rightPos <= rightEnd){
        if(a[leftPos] <= a[rightPos])
            tmpArray[tmpPos++] = a[leftPos++];
        else
            tmpArray[tmpPos++] = a[rightPos++];
    }

    while(leftPos <= leftEnd)
        tmpArray[tmpPos++] = a[leftPos++];

    while(rightPos <= rightEnd)
        tmpArray[tmpPos++] = a[rightPos++];

    for (int i = 0; i < numElements; i++, rightEnd--)
        a[rightEnd] = tmpArray[rightEnd];
}

template<class T>
void mergeSort(vector<T> &a, vector<T> & tmpArray, int left, int right)
{
    if (left < right){
        int center = (left+right)/2;
        mergeSort(a, tmpArray, left, center);
        mergeSort(a, tmpArray, center+1, right);
        merge(a, tmpArray, left, center+1, right);
    }
}

template<class T>
void mergeSort(vector<T> & a)
{
    vector<T> tmpArray(a.size());

    mergeSort(a, tmpArray, 0, a.size()-1);
}

int main()
{
    vector<int> v;
    v.push_back(12);
    v.push_back(8);
    v.push_back(3);
    v.push_back(8);
    v.push_back(10);
    v.push_back(1);
    v.push_back(2);
    v.push_back(14);
    v.push_back(15);
    v.push_back(9);
    v.push_back(7);
    v.push_back(4);

    cout << "v: ";
    copy(v.begin(), v.end(), ostream_iterator<int>(cout, " "));
    cout << endl;

    mergeSort(v);

    cout << "mergeSorted v: ";
    copy(v.begin(), v.end(), ostream_iterator<int>(cout, " "));
    cout << endl;

    return 0;
}



相關文章