昨天腾讯一面上来就给我整的这道 easy 难度的题,然后我太紧张了还想了一会儿,差点炸裂。
题目链接
LeetCode 88. 合并两个有序数组[1]
题目描述
给你两个有序整数数组 nums1
和 nums2
,请你将 nums2
合并到 nums1
中,使 nums1
成为一个有序数组。
说明:
- 初始化
nums1
和nums2
的元素数量分别为m
和n
。 - 你可以假设
nums1
有足够的空间(空间大小大于或等于m + n
)来保存nums2
中的元素。
示例1
题解
看到这道题,脑海里应该第一时间想到的是归并排序,但是归并排序需要一个额外的数组用来保存排序后的数组,这里不允许使用额外空间。
那么我们还是用归并排序的思路来做,试一下两个指针 i = 0
和 j = 0
,初始的时候分别指着 nums1[0]
和 nums2[0]
。然后比较 nums1[i]
和 nums2[j]
大小,如果 nums1[i]
更小,那么就放在原位不动它,然后 i += 1
。如果 nums2[j]
更小,那么就交换 nums1[i]
和 nums2[j]
,然后还是 i += 1
。这么看貌似可行哦?但是最终一定会先遍历完 nums1
,然后 j
还是停留在 0
,然后你会发现 nums2
中的数字还是乱序的,根本没法处理。
那么怎么利用上 nums1
后面多出的那么多空位呢?我们可以换个思路,从最大的开始遍历。两个指针初始的时候 i = m-1
和 j = n-1
,然后将较大值填充到 nums1
的最后面。最后如果 nums2
中还有剩余,就依次填充到 nums1
最前面就行了。
这样为什么就可以了呢?因为如果从小到大遍历的话,元素会覆盖掉 nums1
中还没遍历的元素。但是从大到小是填充到尾部,就不会产生覆盖。就算极限情况下 nums2
中元素全部大于 nums1
中元素,也不会覆盖到 nums1
的最后一个元素。
面试官最后还会问你有啥优化,我当时图省事,最后还把 nums1
中剩下元素填充到 nums1
最前面了,其实完全没有必要,本来就是有序的,等于没有做事。
代码
c++
python
参考资料
[1]
LeetCode 88. 合并两个有序数组: https://leetcode-cn.com/problems/merge-sorted-array/