合并排序的输出不正确

时间:2016-09-19 20:33:06

标签: c++ algorithm sorting merge

我正在尝试使用c ++中的模板函数编写合并排序算法。输出接近但不正确。当我给函数数组[7,6,4,8,1,2,3]时,它返回[1,1,2,2,3,3,8]。我特别相信问题出在合并函数而不是合并排序函数中。任何帮助将非常感激。这是我的代码:

template <class T1>
void mergeSort(T1 array[], int lower, int upper)
{
    if (lower < upper)
    {
        int middle = (lower + upper) / 2;

        mergeSort(array, lower, middle);
        mergeSort(array, middle + 1, upper);
        merge(array, lower, middle, upper);
    }
}

template <class T1>
void merge(T1 array1[], int lower, int middle, int upper)
{
    int i = 0,
        j = 0,
        k = 0;
    int size1 = middle - lower + 1;
    int size2 = upper - middle;
    T1* temp1 = new T1[size1];
    T1* temp2 = new T1[size2];

    for (int i = 0; i < size1; i++)
    {
        temp1[i] = array1[lower + i];
    }
    for (int j = 0; j < size2; j++)
    {
        temp2[j] = array1[middle + 1 + j];
    }

    while (i < size1 && j < size2)
    {
        if (temp1[i] < temp2[j])
        {
            array1[k] = temp1[i];
            i++;
        }
        else
        {
            array1[k] = temp2[j];
            j++;
        }
        k++;
    }

    if (i == size1)
    {
        while (j < size2)
        {
            array1[k] = temp2[j];
            k++;
            j++;
        }
    }
    else
    {
        while (i < size1)
        {
            array1[k] = temp1[i];
            k++;
            i++;
        }
    }
}

int main(){
     int a[] = { 7, 6, 4, 8, 1, 2, 3 };
     mergeSort(a, 0, 6);
}

输出:

1 1 2 2 3 3 8

1 个答案:

答案 0 :(得分:2)

merge函数中,您不应将k初始化为0,因为它会将合并结果写入错误的位置。相反,您应该将k初始化为lower。因为这是您实际排序的部分的起始索引。