Swift 归并排序

Swift 归并排序

基本原理:

对于两个有序子数组，合并成一个有序数组，是一个较为简单的事情。可以对两个子数组，从头开始遍历，并进行比较，如果哪个子数组的元素更需要排到前面，那么就把这个元素排到一个合并后的数组的前面。并移动该子数组的指针，指向下一个元素，合并后的数组也指向下一个元素，再进行下一轮比较。如果哪个数组已经指针移动到最后，那么另外的数组的剩余元素直接添加到合并后数组的当前位置后面。

如何把一个数组拆成两个有序数组，就需要使用递归。通过递归把数组拆成两个有序数组，如果还不是继续递归拆分。然后执行上面的合并逻辑。因为最终可以拆分成1个元素或者0个元素，所以，递归后，总是可以变成有序数组。说可以可以进行合并。

上代码

func mergeSort(arr: inout [Int], mergeTempArr: inout [Int], left: Int, right: Int) {
    
    if left >= right {
        return
    }
    
    let mid = left + (right - left) >> 1
    
    mergeSort(arr: &arr, mergeTempArr: &mergeTempArr, left: left, right: mid)
    mergeSort(arr: &arr, mergeTempArr: &mergeTempArr, left: mid + 1, right: right)
    
    var i = left
    var j = mid + 1
    var k = left
    
    while i <= mid, j <= right {
        if arr[i] < arr[j] {
            mergeTempArr[k] = arr[i]
            i += 1
        } else {
            mergeTempArr[k] = arr[j]
            j += 1
        }
        k += 1
    }
    while i <= mid {
        mergeTempArr[k] = arr[i]
        i += 1
        k += 1
    }
    
    while j <= right {
        mergeTempArr[k] = arr[j]
        j += 1
        k += 1
    }
    
    for index in left...right {
        arr[index] = mergeTempArr[index]
    }
}

var tempArr = arr

mergeSort(arr: &arr, mergeTempArr: &tempArr, left: 0, right: arr.count - 1)

print(arr)