给定两个有序整数数组 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]
题目不难,第一种思路是设一个新的数组,长度为m+n,然后注意的将nums1和nums2数组比较,小的元素放入新的数组中,最后将新的数组copy到nums1中
代码:
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int []nums = new int[m+n];
int i = 0,j = 0,k = 0;
while(i<m && j<n){
if(nums1[i]<=nums2[j]){
nums[k] = nums1[i];
i++;
k++;
}
if(nums1[i]>nums2[j]){
nums[k] = nums2[j];
j++;
k++;
}
}
while(i<m){
nums[k] = nums1[i];
i++;
k++;
}
while(j<n){
nums[k] = nums2[j];
j++;
k++;
}
for(int l = 0;l<k;l++){
nums1[l] = nums[l];
}
}
}
代码比较理解起来比较简单,不多说了,但是是不是还有更简便的算法呢?
如果我们从两个有序数组最后一个元素开始比较,较大的放入nums1数组的m+n-1元素中,
然后索引后移,再将较大的放入m+n-2元素中,这样下去就产生两种情况,
第一种,就是nums1的所有元素被比较完了,直接将nums2里的元素copy到nums1里就可以了
第二种,nums2里的所有元素被比较完了,由于两数组之前都是有序的,所以合并就完成了。
代码:
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int k = m+n-1;
--m;--n;
while(m>=0 && n>=0) nums1[k--] = nums1[m] > nums2[n] ? nums1[m--]:nums2[n--];
while(n>=0) nums1[k--] = nums2[n--];
}
}
后一种方法我也是看到一位大佬的博客写的,一时找不到这位大佬的链接啦,哈哈,在这里谢谢大佬啦