912. 排序数组
原始题目链接:https://leetcode-cn.com/problems/sort-an-array/
给你一个整数数组 nums,请你将该数组升序排列。
示例 1:
输入:nums = [5,2,3,1]
输出:[1,2,3,5]
示例 2:
输入:nums = [5,1,1,2,0,0]
输出:[0,0,1,1,2,5]
解题思路:
排序的基础题,手写快排,归并,堆排序。
代码实现:
快速排序
class Solution:
def sortArray(self, nums: List[int]) -> List[int]:
def helper(nums, left, right):
pivot_index = random.randint(left, right)
nums[pivot_index], nums[left] = nums[left], nums[pivot_index]
i, j = left, right
pivot = nums[left]
while i < j:
while i < j and pivot <= nums[j]:
j -= 1
nums[i] = nums[j]
while i < j and pivot >= nums[i]:
i += 1
nums[j] = nums[i]
nums[i] = pivot
return i
def quick_sort(nums, left, right):
if left >= right:
return
index = helper(nums, left, right)
quick_sort(nums, left, index - 1)
quick_sort(nums, index + 1, right)
quick_sort(nums, 0, len(nums) - 1)
return nums
归并排序
class Solution:
def sortArray(self, nums: List[int]) -> List[int]:
# 定义一个函数,用于归并排序
def merge_sort(array, left, right):
# 递归的结束条件,当元素都排序完的时候
if left >= right:
return
# 归并排序的分的阶段,就是一分为二
mid = left + (right - left) // 2
# 递归的对一分为二的两边继续使用归并排序
merge_sort(array, left, mid)
merge_sort(array, mid + 1, right)
# 定义两个指针,分别指向待排序数组的左右半边的开始位置
i, j = left, mid + 1
# 存储一趟的归并排序结果,最后赋值给原数组
temp = []
# 开始治的阶段,其实就是比较排序
while i <= mid and j <= right:
# 添加较小者,就是从大到小排序
if array[i] <= array[j]:
temp.append(array[i])
i += 1
else:
temp.append(array[j])
j += 1
# 上述比较,可能存在左右半边剩余还没插入temp数组的情况
# 将剩余的元素插入temp数组即可
temp += array[i : mid + 1]
temp += array[j : right + 1]
# 将完成一次排序的数组结果分配给原数组
array[left : right + 1] = temp
merge_sort(nums, 0, len(nums) - 1)
return nums