终于还是踏上了算法这条不归路,记录一下做题的历程
合并两个有序数组
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化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]输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
直接合并后排序
很直观,先将数组2合并到数组1中,再对数组1进行排序,代码也很简单。
代码
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
for(int i = 0; i<n ;++i){ // 先合并
nums1[m+i] = nums2[i];
}
sort(nums1.begin(),nums1.end()); // 在排序
}
};
逆向双指针
从尾部开始比较遍历,将较大值填进nums1末位中。
原nums1中的元素比较完后,还需要将nums2中剩余元素依次保存至nums1中。
代码
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int len1 = m - 1;
int len2 = n - 1;
int len = m + n - 1;
while(len1 >= 0 && len2 >= 0) { // 两数组均有元素时进行比较(从后向前),较大值放入nums1末位。
// nums1[len--] = nums1[len1] > nums2[len2] ? nums1[len1--] : nums2[len2--]; // 下面代码的简化版
if(nums1[len1]> nums2[len2]){ // 从后向前比较两数组各自最大值的较大值,较大值存入末位。
nums1[len] = nums1[len1];
len--;
len1--;
}
else{
nums1[len] = nums2[len2];
len--;
len2--;
}
}
while(len1<0 && len2>=0){ // 当len1<0时,若len2中还有元素,将len2中的元素(有序数组)依次保存至len1中。
nums1[len2] = nums2[len2];
len2--;
}
}
};
参考
作者:guanpengchn
链接:https://leetcode-cn.com/problems/merge-sorted-array/solution/hua-jie-suan-fa-88-he-bing-liang-ge-you-xu-shu-zu-/
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-sorted-array