题目描述:
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n。你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
解题思路1:
将nums2中的元素插到nums1后面,然后对nums1进行排序
代码1:
class Solution(object):
def merge(self, nums1, m, nums2, n):
i = m
while i < len(nums1):
for j in range(n):
nums1[i] = nums2[j]
i += 1
return sorted(nums1)
s = Solution()
nums1 = [1, 2, 3, 0, 0, 0]
nums2 = [2, 5, 6]
m = 3
n = 3
print(s.merge(nums1, m, nums2, n))
class Solution(object):
def merge(self, nums1, m, nums2, n):
for index, item in enumerate(nums2):
nums1[m+index] = item
nums1.sort()
return nums1
s = Solution()
nums1 = [1, 2, 3, 0, 0, 0]
nums2 = [2, 5, 6]
m = 3
n = 3
print(s.merge(nums1, m, nums2, n))
class Solution(object):
def merge(self, nums1, m, nums2, n):
for i in range(n):
nums1[m] = nums2[i]
m += 1
return sorted(nums1)
解题思路2: 双指针解法 / 滑动窗口解法
- 初始时,指针 i i i 指向第一个数组有效空间的末尾位置,指针 j j j指向第二个数组的末尾,指针 t t t 指向第一个数组的末尾。
- 指针 i i i 与指针 j j j 所指的数字进行比较,将较大的数字存放进指针 t t t所指向的位置,然后将 i i i 与 j j j 中指向较大的指针向前移动一位, t t t 也向前移动一位。
- 当第一个指针移动完毕而第二个指针仍有剩余时,将剩余数字写入数组num1中去。
代码2:
class Solution(object):
def merge(self, nums1, m, nums2, n):
t = m + n - 1
i = m-1
j = n-1
while i >= 0 and j >= 0:
if nums1[i] >= nums2[j]:
nums1[t] = nums1[i]
i -= 1
else:
nums1[t] = nums2[j]
j -= 1
t -= 1
while j >= 0:
nums1[t] = nums2[j]
t -= 1
j -= 1
return nums1
另一种双指针的走法:
i i i 指向的是nums1的初始位置, j j j指向的是nums2的初始位置。通过m, n来约束界限。
class Solution:
def merge(self, nums1, m, nums2, n):
nums = []
i, j = 0, 0
while i < m and j < n:
if nums1[i] < nums2[j]:
nums.append(nums1[i])
i += 1
else:
nums.append(nums2[j])
j += 1
while i < m: # 哪个有剩余,哪个将剩下的元素追加进nums
nums.append(nums1[i])
i += 1
while j < n:
nums.append(nums2[j])
j += 1
return nums