题目描述
【leetcode】88. 合并两个有序数组( Merge Sorted Array )
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
第一次解答
思路:
nums1已经有足够的空间,
数组从小到大排序
为了防止大量的内存拷贝,从后往前添加元素
test case:
[1,2,3,0,0,0]
3
[2,5,6]
3
[4,5,6,0,0,0]
3
[1,2,3]
3
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
// 序号指向两个数组末尾
int index_1 = m-1;
int index_2 = n-1;
for(int i=m+n-1; i>=0; --i){
// 如果一个数组序号小于0,那只添加另一个数组元素
if(0 > index_1){
nums1[i] = nums2[index_2--];
continue;
}
if(0 > index_2){
nums1[i] = nums1[index_1--];
continue;
}
// 若序号都大于0,挑选数值大的元素添加
if(nums1[index_1] > nums2[index_2]){
nums1[i] = nums1[index_1--];
}
else{
nums1[i] = nums2[index_2--];
}
}
}
};
结果:
第二次解答
思路:
方法与第一次解答一样,不变,优化代码减少if判断次数。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
// 序号指向两个数组末尾
int index_1 = m-1;
int index_2 = n-1;
int index = m+n-1;
while(0 <= index_1 && 0 <= index_2){
// 若序号都大于0,挑选数值大的元素添加
nums1[index--] = (nums1[index_1] > nums2[index_2] ? nums1[index_1--] : nums2[index_2--]);
}
// 到这里后,若0 > index_2,则merge完成
// 若0 > index_1,则把nums2剩下元素放到nums1中即可
while(0 <= index_2){
nums1[index--] = nums2[index_2--];
}
}
};
结果: