You are given two integer arrays nums1
and nums2
, sorted in non-decreasing order, and two integers m
and n
, representing the number of elements in nums1
and nums2
respectively.
Merge nums1
and nums2
into a single array sorted in non-decreasing order.
The final sorted array should not be returned by the function, but instead be stored inside the array nums1
. To accommodate this, nums1
has a length of m + n
, where the first m
elements denote the elements that should be merged, and the last n
elements are set to 0
and should be ignored. nums2
has a length of n
.
Example 1:
Input: nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3 Output: [1,2,2,3,5,6] Explanation: The arrays we are merging are [1,2,3] and [2,5,6]. The result of the merge is [1,2,2,3,5,6] with the underlined elements coming from nums1.
Example 2:
Input: nums1 = [1], m = 1, nums2 = [], n = 0 Output: [1] Explanation: The arrays we are merging are [1] and []. The result of the merge is [1].
Example 3:
Input: nums1 = [0], m = 0, nums2 = [1], n = 1 Output: [1] Explanation: The arrays we are merging are [] and [1]. The result of the merge is [1]. Note that because m = 0, there are no elements in nums1. The 0 is only there to ensure the merge result can fit in nums1.
Constraints:
nums1.length == m + n
nums2.length == n
0 <= m, n <= 200
1 <= m + n <= 200
-109 <= nums1[i], nums2[j] <= 109
Follow up: Can you come up with an algorithm that runs in O(m + n)
time?
The problem asks to merge two sorted arrays, nums1
and nums2
, into nums1
in a sorted manner. nums1
initially contains m
elements, followed by n
zeros to accommodate the elements from nums2
.
The most efficient approach uses a two-pointer technique. This avoids the need for extra space and achieves linear time complexity.
Initialization: We start with three pointers:
k
: Points to the end of nums1
(index m + n - 1
). This is where we'll place the merged elements.i
: Points to the end of the valid portion of nums1
(index m - 1
).j
: Points to the end of nums2
(index n - 1
).Iteration: We iterate while j
(pointer for nums2
) is within the bounds of nums2
. In each iteration:
nums1[i]
and nums2[j]
.nums1[k]
.i
or j
).k
is always decremented to move to the next position in nums1
.Handling Remaining Elements: After the loop, if i
is still within nums1
's valid bounds, then it implies that there are some elements in nums1
that haven't been considered. However, since nums1
is already sorted, these elements are already in the correct position. If any elements remain in nums2
, they are already smaller than all elements in the remaining portion of nums1
. Therefore, no further action is needed.
Python:
class Solution:
def merge(self, nums1: list[int], m: int, nums2: list[int], n: int) -> None:
"""
Merges nums2 into nums1 in-place using a two-pointer approach.
"""
# Pointers for nums1, nums2, and the merged array
k = m + n - 1 # End of nums1 (merged array)
i = m - 1 # End of valid portion of nums1
j = n - 1 # End of nums2
# Iterate until we've processed all elements in nums2
while j >= 0:
if i >= 0 and nums1[i] > nums2[j]:
nums1[k] = nums1[i] # Place larger element from nums1
i -= 1 # Move pointer for nums1
else:
nums1[k] = nums2[j] # Place larger element from nums2
j -= 1 # Move pointer for nums2
k -= 1 # Move pointer for merged array
Java:
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int k = m + n - 1; // index of the last element in nums1 (merged array)
int i = m - 1; // index of the last element in the valid portion of nums1
int j = n - 1; // index of the last element in nums2
while (j >= 0) {
if (i >= 0 && nums1[i] > nums2[j]) {
nums1[k--] = nums1[i--];
} else {
nums1[k--] = nums2[j--];
}
}
}
}
The code in other languages (C++, Go, TypeScript, Rust, Javascript, PHP) follows a very similar structure, employing the same two-pointer approach. The core logic remains consistent across all implementations.