编程语言
首页 > 编程语言> > [每日一水] 归并排序 Python实现

[每日一水] 归并排序 Python实现

作者:互联网

(建议先看后边的伪代码分析,然后自己写出来)


先上代码:

import numpy as np


length = 100000
array = np.arange(0, length)
np.random.shuffle(array)


def merge(array, s1, e1, s2, e2):
    array_new = array.copy()
    # 规定从小到大
    
    steps = (e1-s1+1) + (e2-s2+1)
    start = s1
    for i in range(steps):
        
        if s1 > e1:
            array_new[start+i] = array[s2]
            s2 += 1
            continue
            
        if s2 > e2:
            array_new[start+i] = array[s1]
            s1 += 1
            continue
        
        if   array[s1] >= array[s2]:
            array_new[start+i] = array[s2]
            s2 += 1
            
        elif array[s1]  < array[s2]:
            array_new[start+i] = array[s1]
            s1 += 1
    
    array[start:e2+1] = array_new[start:e2+1]
    
    return array_new
        


def mergeSort(array, start, end):
    # 规定从小到大
    
    if end-start == 1:
    	
        if array[start] > array[end]:
            array[start] += array[end]
            array[end] = array[start] - array[end]
            array[start] = array[start] - array[end]
        return array
    
    if end == start:
        return array
    
    mid   = int((start + end) / 2)
    
    mergeSort(array, start, mid)
    mergeSort(array, mid+1, end)
    
    merge(array, start, mid, mid+1, end)

print(array)    
mergeSort(array, 0, length-1)
print(array)

俺是非科班出生的菜鸡,一大把年纪了,第一次写归并排序

标签:mergeSort,归并,end,Python,s2,s1,start,一水,array
来源: https://blog.csdn.net/HaoZiHuang/article/details/120589506