将两个有序数组合并成一个有序数组,合并到第一个数组中。
初始化num1和num2的元素数量分别为m和n。假设num1空间大小等于m+n,这样就有足够的空间保存来自nums2的元素。
利用java的api:先将num2拷贝到num1中,然后再将num1进行排序。时间复杂度O((n+m)log(n+m)),没有引入额外空间。
双指针算法:设置两个指针,分别指向两个数组起始位置。每次比较当前两个数字大小,小的那个就进入新的数组,然后指针后移。由于要返回num1,所以不妨将num1拷贝到新数组,将num1作为结果数组返回。注意要考虑某个数组遍历完而另一个数组未变遍历完的情况即可。时间和空间复杂度O(m+n)。
双指针算法优化:我们可以从后往前比较。由于num1后面几位都是占位符,因此我们只需要将比较结果较大的那一个放进num1尾部即可。这样就不需要额外空间。
上代码:
import java.lang.reflect.Array;
import java.util.Arrays;
//12.合并有序数组
public class MergeSortArray {
public static void main(String[] args) {
int[] num1 = new int[]{2,4,6,8,10,0,0,0,0};
int[] num2 = new int[]{1,3,5,7};
// System.out.println(Arrays.toString(merge(num1,5,num2,4)));
// System.out.println(Arrays.toString(merge1(num1,5,num2,4)));
System.out.println(Arrays.toString(merge3(num1,5,num2,4)));
}
//利用java中的api
private static int[] merge(int[] num1, int m, int[] num2, int n) {
System.arraycopy(num2,0,num1,m,n);
Arrays.sort(num1);
return num1;
}
//双指针算法
private static int[] merge1(int[] nums1, int m, int[] nums2, int n) {
int[] num1_copy = new int[m];//新的数组
System.arraycopy(nums1,0,num1_copy,0,m);//为了不违背题意
int p1=0,p2=0;//双指针
int p = 0;//指向num1。
while (p1<m&&p2<n){
nums1[p++] = num1_copy[p1]<nums2[p2]?num1_copy[p1++]:nums2[p2++];
}
if (p1<m){
System.arraycopy(num1_copy,p1,nums1,p1+p2,m+n-p1-p2);
}
if (p2<n){
System.arraycopy(nums2,p2,nums1,p1+p2,m+n-p1-p2);
}
return nums1;
}
//双指针算法优化
private static int[] merge3(int[] num1, int m, int[] num2, int n) {
int p1 = m-1;
int p2 = n-1;
int p = num1.length-1;//num1的最后一个下标
while (p1>=0&&p2>=0){
num1[p--] = num1[p1]<num2[p2]?num2[p2--]:num1[p1--];
}
System.arraycopy(num2,0,num1,0,p2+1);
return num1;
}
}
结果均为:(注意必须一次运行一个,因为是引用,前面的结果会影响后面,每次测试时必须注释掉其他的测试)
[1,2,3,4,5,6,7,8,10]