问题:
给你两个有序整数数组 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]
分析:
今天学习到了归并排序,得到启发,可以使用一个新的数组,并定义两个索引,从头遍历两个数组,两个索引指向的元素比较小的填入新数组,索引向后位移一位,再次比较,直到某一索引走至元素最后一位,若另一个数组中的索引并未走至最后一位,则依次将数据填入新数组最后元素的后面,并将新数组中的元素的值,赋给num1这个数组,即可完成此题。完成后,通过思考,选择从尾部插入的方法解决此题。
第一种方法:
public void merge(int[] nums1, int m, int[] nums2, int n) {
if(m==0){
for (int i = 0; i < nums2.length; i++) {
nums1[i]=nums2[i];
}
return;
}else if(n==0){
return;
}
int[] arr=new int[m+n];
int a=0;
int b=0;
for (int i = 0; i <arr.length; i++) {
if(nums1[a]>nums2[b]) {
arr[i] = nums2[b];
b++;
}else if(nums1[a]<nums2[b]){
arr[i]=nums1[a];
a++;
}else if(nums1[a]==nums2[b]){
arr[i]=nums1[a];
arr[i+1]=nums2[b];
a++;
b++;
i++;
}
if(a==m&&b<=n){
for (int j = b; j < n; j++) {
arr[i+1+j-b]=nums2[j];
}
break;
}if(b==n&&a<=m){
for (int j = a; j <m; j++) {
arr[i+1+j-a]=nums1[j];
}
break;
}
}
for (int i = 0; i <arr.length; i++) {
nums1[i]=arr[i];
}
}
第二种方法:
因为第一个数组,最后一个元素后部,全部为默认值0,我们可以通过m+n找到我们想得到的结果的数组最后一个索引的位置,然后在两个数组中定义两个索引,从第m和n个位置开始,向前遍历,并比较大小,将大的填入第一个数组的最后部分,并将小的索引向前移动一位。,若第一个数组的索引先走到头时,我们将第二个数组从头部插入第一个数组。若第二个数组的索引先走到头,我们则结束此次排序,第一个数组就是我们所需要的数组。
代码如下:
public void merge(int[] nums1, int m, int[] nums2, int n) {
if(m==0){
for (int i = 0; i < nums2.length; i++) {
nums1[i]=nums2[i];
}
return;
}else if(n==0){
return;
}
int a=m-1;
int b=n-1;
for (int i = m+n-1; i >=0; i--) {
if(nums1[a]>nums2[b]){
nums1[i]=nums1[a];
a--;
}else if(nums1[a]<nums2[b]){
nums1[i]=nums2[b];
b--;
}else if(nums1[a]==nums2[b]){
nums1[i]=nums1[a];
a--;
nums1[i-1]=nums2[b];
i--;
b--;
}
if(b<0){
return;
}
if(a<0){
for (int j = i; j>0; j--) {
nums1[j-1]=nums2[j-1];
}
return;
}
}
}