归并排序

归并排序

通过阅读《算法导论》第二章，学习了算法分析相关内容。其中，包含了插入排序、冒泡排序和归并排序。

本文主要列出归并排序算法执行步骤和C++、Python代码。

算法名次	归并排序
类别	排序算法
复杂度	nlgn
实现形式	数组
参考书	算法导论

归并排序步骤：
１：将数组对半分成两组，对两组重复执行步骤1，直到分组后只剩下一个元素为止；
２：将两个对分元素进行排序，然后再向上（返回）重复进行步骤2，直到最上层数组完成排序；

C++代码

#include <iostream>

/////////////////////
//函数名：merge
//函数功能：对两组待排序数组执行合并排序
//输入变量：
// merge_array：数组，存放待排序数组
// first：整型，待排序数组第一组首数据下标
// middle：整型，待排序数组第一组尾数据下标
// end：整型，待排序数组第二组尾数据下标
//输出变量：无
/////////////////////
void merge(int merge_array[], int first, int middle, int end)
{
  int left_num = middle - first + 1;
  int right_num = end - middle;

  int *left_array = new int[left_num];
  int *right_array = new int[right_num];

  for(int i = first, j = 0; i <= middle; ++i, ++j)
    left_array[j] = merge_array[i];

  for(int i = middle + 1, j = 0; j <= end; ++i, ++j)
    right_array[j] = merge_array[i];

  for(int key = first, left_index = 0, right_index = 0; key <= end; ++key)
  {
    if(left_index == left_num)
    {
      while(key <= end)
        merge_array[key++] = right_array[right_index++];
      break;
    }
    else if(right_index == right_num)
    {
      while(key <= end)
        merge_array[key++] = left_array[left_index++];
      break;
    }
    else
    {
      if(left_array[left_index] < right_array[right_index])
        merge_array[key] = left_array[left_index++];
      else
        merge_array[key] = right_array[right_index++];
    }
  }

  delete []left_array;
  delete []right_array;
}

/////////////////////
//函数名：merge_sort
//函数功能：对两组待排序数组执行分组、归并
//输入变量：
// merge_array：数组，存放待排序数组
// first：整型，待排序数组首数据下标
// end：整型，待排序数组尾数据下标
//输出变量：无
/////////////////////
void merge_sort(int merge_array[], int first, int end)
{
  if(first < end)
  {
    int middle = (first + end) /2;

    merge_sort(merge_array, first, middle);
    merge_sort(merge_array, middle + 1, end);

    merge(merge_array, first, middle, end);
  }

}

int main()
{
  int array[] = {234, 1, 45, 67, 21, 975, 5275, 867, -12, -200};

  int num = 10;

  printf("待排序数组顺序为：");
  for(int i = 0; i < num; ++i)
    printf("%d ", array[i]);

  printf("\n");

  merge_sort(array, 0, num - 1);

  printf("已排序序数组顺序为：");
  for(int i = 0; i < num; ++i)
    printf("%d ", array[i]);


  printf("\nFinished.\n");

  return 0;
}

执行结果：

待排序数组顺序为：234 1 45 67 21 975 5275 867 -12 -200
已排序序数组顺序为：-200 -12 1 21 45 67 234 867 975 5275
Finished.
Press any key to continue…

Python代码

####################
#函数名：merge
#函数功能：对两组待排序数组执行合并排序
#输入变量：
# merge_array：数组，存放待排序数组
# first：整型，待排序数组第一组首数据下标
# middle：整型，待排序数组第一组尾数据下标
# end：整型，待排序数组第二组尾数据下标
#输出变量：无
####################
def merge(merge_array, first, middle, end):
    left_num = middle - first + 1
    right_num = end - middle

    left_array = merge_array[first : middle + 1]
    right_array = merge_array[middle + 1 : end + 1]

#    print("left num", left_num, first, middle)
#    print("left:", left_array)
#    print("right num", right_num, middle + 1, end)
#    print("right", right_array)

    left_index = 0
    right_index = 0

    for key in range(first, end + 1):
        if left_index == left_num:
            merge_array[key : end + 1] = right_array[right_index :]
            break
        elif right_index == right_num:
            merge_array[key : end + 1] = left_array[left_index :]
            break
        else :
            if left_array[left_index] < right_array[right_index]:
                merge_array[key] = left_array[left_index]
                left_index = left_index + 1
            else:
                merge_array[key] = right_array[right_index]
                right_index = right_index + 1


####################
#函数名：merge_sort
#函数功能：对两组待排序数组执行分组、归并
#输入变量：
# merge_array：数组，存放待排序数组
# first：整型，待排序数组首数据下标
# end：整型，待排序数组尾数据下标
#输出变量：无
####################
def merge_sort(merge_array, first, end):

    if first < end:
        middle = int((first + end) / 2)
        merge_sort(merge_array, first, middle)
        merge_sort(merge_array, middle + 1, end)
        merge(merge_array, first, middle, end)



if __name__ == '__main__':
    array = [234, 1, 45, 67, 21, 975, 5275, 867, -12, -200]

    print("待排序数组为：", array)

    merge_sort(array, 0, 9)

    print("已排序数组为：", array)

执行结果：

python3 insertion_sort.py
待排序数组为： [234, 1, 45, 67, 21, 975, 5275, 867, -12, -200]
已排序数组为： [-200, -12, 1, 21, 45, 67, 234, 867, 975, 5275]