题088.合并两个有序数组
题意
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 有足够的空间(空间大小等于 m + n)来保存 nums2 中的元素。
eg
输入: nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
解题
对于本题,最直接的方法就是利用pyhon的排序功能,把nums2塞到nums1后面,直接排序,不过这样就浪费了题中两个有序数组的条件。
所以还是要 利用好有序数组这个信息 ,可以有两种方法
-
一种从前面开始,每次取两个数组头更小的值,因为题目要求全部放到nums1中,但真的直接放nums1中的话还要考虑将后续数组后移,太麻烦,所以将答案先存到临时数组中,最后令nums1等于临时数组,这样就增加了空间复杂度。
-
另一种从后面开始,每次取两个数组尾更大的值,塞到nums1尾部,因为题中已经说了,nums1有足够的空间,所以不要考虑数组的移位问题,也不需要临时数组,明显这种方法更好一点。
解题思路:
- 把更小的数放到 nums1 尾部,直至 nums1 或 nums2 的指针移动到头部
- 判断当前哪个数组有剩余
- 如果 nums1 有剩余,无需操作
- 如果 nums2 有剩余,把它覆盖到 nums1 等长的头部
while m and n:
if nums1[m-1]>nums2[n-1]:
nums1[m+n-1]=nums1[m-1]
m-=1
else:
nums1[m+n-1]=nums2[