给定两个有序整数数组 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]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1:
使用优先队列priority_queue 将nums2中的元素构造成一个小顶堆minHeap,然后将每次用堆顶元素与nums1中的元素依次比较,如果堆顶元素小就将两个元素互换位置:堆顶元素加入到当前nums1中元素的位置,并将nums1中的元素压入堆中,重新构造小顶堆。如此一直到nums1中m的位置,从此之后只是将小顶堆的堆顶元素依次加入nums1即可。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
if (n == 0)
return ;
priority_queue<int, vector<int>, greater<int>> minHeap;
for (int i = 0; i < n; i++)
minHeap.push(nums2[i]);
int ptr = 0;
while (ptr < m) {
if (minHeap.top() < nums1[ptr]) {
int tmp = nums1[ptr];
nums1[ptr] = minHeap.top();
minHeap.pop();
minHeap.push(tmp);
}
ptr++;
}
while (!minHeap.empty()) {
nums1[ptr++] = minHeap.top();
minHeap.pop();
}
}
};
方法2:
双指针 / 从后往前
将指针p 置为 nums的(m+n-1)
处,将指针p1 置为 nums1的(m)
处, p2为 nums2的(n)
处,在每一步将最大值放入nums1中p处。
注意:
如果当nums2倒叙遍历完毕(p2 = -1 < 0),此时nums1还未结束(p1 >= 0 && p >= 0),则剩余元素已经有序的存在nums1中,结束即可;
如果当nums1倒叙遍历完毕(p1 = -1 < 0),此时nums2还未结束(p2 = p >= 0),则将nums2中剩余元素(已经有序)复制到nums1中p之后的空间,结束即可;
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
if (n == 0)
return ;
if (m == 0) {
nums1 = nums2;
return ;
}
int p1 = m - 1;
int p2 = n - 1;
int p = m + n - 1;
while (p1 >= 0 && p2 >= 0)
nums1[p--] = nums1[p1] >= nums2[p2] ? nums1[p1--] : nums2[p2--];
while (p2 >= 0){
nums1[p2] = nums2[p2];
p2--;
}
}
};