Mergesort
LeetCode Problem
# https://leetcode.com/problems/sort-an-array/
class Solution:
def sortArray(self, nums: List[int]) -> List[int]:
def merge(arr, L, M, R):
left, right = arr[L : M + 1], arr[M + 1 : R + 1]
i, j, k = L, 0, 0 # compare j and k
while j < len(left) and k < len(right):
if left[j] <= right[k]:
arr[i] = left[j]
j += 1
else:
arr[i] = right[k]
k += 1
i += 1 # shift regardless
while j < len(left):
nums[i] = left[j]
j += 1
i += 1
while k < len(right):
nums[i] = right[k]
k += 1
i += 1
def mergeSort(arr, l, r):
if l == r:
return arr
mid = (l + r) // 2
mergeSort(arr, l, mid)
mergeSort(arr, mid + 1, r)
merge(arr, l, mid, r)
return arr
return mergeSort(nums, 0, len(nums) - 1)