具有重复键的Quicksort算法

时间:2012-01-26 21:19:59

标签: c++ algorithm sorting quicksort

我正在尝试实施快速排序算法。以下代码适用于唯一元素,但它不适用于具有重复元素的数组。请告诉我我做错了什么。此外,当我将数值的值更改为0以外的其他数字时,程序崩溃。这是代码:

#include <iostream>
#include <cstdlib>

using namespace std;

void swapme(int &a, int &b)
{
    int temp = a;
    a = b;
    b = temp;
}

void quicksort(int *arr, int size)
{    
    // these two variables will take care of position of comparison
    int lower = 0, upper = size - 1;  
    int pivot = 0;  // assigns pivot
    if (size <= 1)
        return;

    while (lower < upper)
    {
        while (arr[lower] < arr[pivot])
        {
            ++lower;
        }
    }

    while (arr[upper] > arr[pivot])
    {
        --upper;
    }

    if (upper > lower)
    {
        swapme(arr[upper], arr[lower]);
        // upper--;
        // lower++;
    }

    quicksort(arr, lower);
    quicksort(&arr[lower + 1], size - 1 - lower);
}

int main()
{
    int arr[30];

    for(int j = 0; j < 30; j++)
    {
        arr[j] = 1 + rand() % 5000;
    }

    for(int j = 0; j < 30; j++)
    {
        cout << arr[j] << "\t";
    }
    cout << endl;

    quicksort(arr, 30);

    for(int j = 0; j < 30; j++)
    {
        cout << arr[j] << "\t";
    }
    cout << endl;

    cin.get();
    cin.get();
}

更新:我终于成功了。这是固定版本:

void swapme(int &a, int &b )
{
    int temp = a;
    a = b;
    b = temp;
}

void quicksort(int *arr, int size)
{
    if (size <= 1)
        return;

    // These two variables will take care of position of comparison.
    int lower = 0;
    int upper = size-1;  

    int pivot = arr[upper/2]; // assigns pivot

    while (lower <= upper)
    {
        while (arr[lower] < pivot)
            ++lower;
        while (arr[upper] > pivot)
            --upper;

        if (upper >= lower)
        {
            swapme(arr[upper],arr[lower]);
            if(arr[upper] == arr[lower])
            {
                // Can either increment or decrement in case of duplicate entry
                upper--; // lower++;
            }
        }
    }

    quicksort(arr, lower);
    quicksort( &arr[lower+1], size-1-lower);
}

2 个答案:

答案 0 :(得分:4)

您正在将{pivot}元素的索引存储在pivot变量中,因此交换元素可能会在循环期间更改pivot元素的选择。不是一个好主意。我建议将pivot元素的实际存储在pivot中。

另外,如果这不是家庭作业,为什么不简单地使用标准的图书馆设施?

#include <algorithm>

// ...

std::sort(arr + 0, arr + 30);

您将获得经过大量优化和测试的代码,这些代码将随时优于您的手写Quicksort。

答案 1 :(得分:0)

可以实现任意数量的i / p整数的快速排序。它还处理重复的密钥

#include <conio.h>
#include <string>

using namespace std;
void InputArray(int*,int);
void QuickSort(int *,int,int);
int partition(int *,int,int);
void swap(int *,int,int);
void printArr(int *,int Siz=11);

void main(){

    int siz;
    cout<<"Enter Array length   :   "; cin>>siz; 
    int *a=new int[siz];

    InputArray(a,siz);
    QuickSort(a,0,siz-1);
    int i=0,j=11;

    printArr(a,siz);
    system("pause");

}


void InputArray(int*a,int s){
    for(int i=0; i<s; i++){
        cout<<"ELement ["<<i<<"] =  "; cin>>a[i];
    }

}
void QuickSort(int *a,int start,int end){
    if(start<end){
        int pivot=partition(a,start,end);

        QuickSort(a,start,pivot);
        QuickSort(a,pivot+1,end);
    }


}
int partition(int *a,int start,int end){

    int currentPivotValue=a[start];
    int i=start-1, j=end+1;
    while(true){
        i++;
        while(i<j && a[i]<currentPivotValue){ i++; }
        j--;

        while(j>start && a[j]>currentPivotValue) {j--;}
        if(i<j) swap(a,i,j);        
        else return j;
    }


}
void swap(int *b,int i,int j){
    int t=b[i];
    b[i]=b[j];
    b[j]=t;



}
void printArr(int *a,int Siz){
    for(int i=0; i<Siz; i++) cout<<a[i]<<" "; 
}