题目描述:
给你两个按 非递减顺序 排列的整数数组 nums1
和 nums2
,另有两个整数 m
和 n
,分别表示 nums1
和 nums2
中的元素数目。
请你 合并 nums2
到 nums1
中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1
中。为了应对这种情况,nums1
的初始长度为 m n
,其中前 m
个元素表示应合并的元素,后 n
个元素为 0
,应忽略。nums2
的长度为 n
。
思路:
很明显的一个感觉就是使用两个指针i,j指向两个数组的末尾,再使用一个指针p标记nums1的实际长度。因为要使得数组非降序排列,所以就开始从最后跑指针,将大的放在p所指的位置,然后使得较大的那个数的指针和p一起往前跑,并且跑完之后由于两个数组本来就是有序的,且i一定比j大,如果j跑完了,那么剩下的i本来就在前面,就不用跑了,所以while循环的终止条件就只要j>0即可。如果i跑完了但是j没有跑完,那么只要将剩下的j继续覆盖即可,因为这时剩下的j指的任意元素都比i原来位置的小。
由以上思路得到题解:
代码语言:python代码运行次数:0复制class Solution:
def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
i = m - 1
j = n - 1
p = m n -1
while j >= 0:
if i >= 0 and nums1[i] > nums2[j]:
nums1[p] = nums1[i]
i -= 1
else:
nums1[p] = nums2[j]
j -= 1
p -= 1