C++归并排序

发布于:2025-08-09 ⋅ 阅读:(18) ⋅ 点赞:(0)

1 算法核心思想

归并排序是一种高效的排序方式,需要用到递归来实现,我们先来看一下动图演示:

算法核心思想如下:

1.将数组尽量平均分成两段。

2.将这两段都变得有序(使用递归实现)。

3.将两段合并。

2 代码实现

首先,我们先定义一个归并排序的函数,里面接受三个参数:

void MergeSort(int arr[], int left, int right) {
    
}

arr代表需要进行排序的数组,left表示数组arr的最左端点,right表示数组arr的最右端点。

首先我们需要把数组分成两段,我们可以用二分的方法:

int mid = (left + right) >> 1;

这里右移(>>为右移运算符)1为和除以2含义相同。

也可以用防溢出,因为left+right的值可能会爆int,导致结果错误:

int mid = left + (right - left) >> 1;

然后对两段分别进行递归,第一段是[1, mid],第二段是[mid+1, right]:

MergeSort(arr, left, mid);
MergeSort(arr, mid + 1, right);

由于我们需要对数组进行操作,但是直接在arr操作可能会导致原始数据丢失,但是如果再创建一个数组会占用内存,所以我们可以向电脑“租借”right-left+1个空间,用关键字new来完成:

int* tmp = new int[right - left + 1];

注意要以指针的形式定义。

由于我们要把数组变得有序,而我们归并排序的思想就是分而治之,然后再依次变得有序,需要用到分治的思想。那么我们先定义一些变量:

int cur = 0, cur1 = left, cur2 = mid + 1;

cur为tmp数组的元素下标,cur1为第一段的最左端点,cur2为第二段的最左端点。

然后我们对tmp数组和arr数组进行循环操作,这里可以用while循环,循环条件是cur1<=mid&&cur2<=right。

如果arr[cur1]比arr[cur2]更大,那么就先把arr[cur2]放回tmp,否则放arr[cur1]。

代码:

while(cur1 <= mid && cur2 <= right)
{
    if(arr[cur1] < arr[cur2])
        tmp[cur++] = arr[cur1++];
    else
        tmp[cur++] = arr[cur2++];
}

然后处理可能有的数组残余未处理的部分:

while(cur1 <= mid)
    tmp[cur++] = arr[cur1++];
while(cur2 <= right)
    tmp[cur++] = arr[cur2++];

然后合并数组,方法跟处理时差不多的:

for(int i = 0; i < right - left + 1; i++)
    arr[left + i] = tmp[i];

就是把tmp的元素依次赋值给arr。

最有我们需要把tmp的空间还给内存,所以我们delete一下:

delete[] tmp;

然后我们的arr就变的有序了。

但是,如果这样写,程序就成功被我们干崩了,因为我们忘记写递归出口了,补一个递归出口:

if(left == right)
    return;

我们合并一下整段代码:

void MergeSort(int arr[], int left, int right) {
    if(left == right)
        return;
    int mid = (left + right) >> 1;
    MergeSort(arr, left, mid);
    MergeSort(arr, mid + 1, right);
    int* tmp = new int[right - left + 1];
    int cur = 0, cur1 = left, cur2 = mid + 1;
    while(cur1 <= mid && cur2 <= right)
    {
        if(arr[cur1] < arr[cur2])
            tmp[cur++] = arr[cur1++];
        else
            tmp[cur++] = arr[cur2++];
    }
    while(cur1 <= mid)
        tmp[cur++] = arr[cur1++];
    while(cur2 <= right)
        tmp[cur++] = arr[cur2++];
    for(int i = 0; i < right - left + 1; i++)
        arr[left + i] = tmp[i];
    delete[] tmp;
}

3 算法时间复杂度

正常情况下,归并排序时间复杂度为:

O(NLogN)

再见!


网站公告

今日签到

点亮在社区的每一天
去签到