这道题有三种解法:
第一种:直接将两个数组合并,再用JDK自带的快排对数组进行排序。
第二种:创建一个临时数组,该数组大小为两个要合并的数组中元素个数之和;利用双指针分别对两个数组进行遍历,在遍历的过程中依次将较小的元素放入临时数组,最后将临时数组的元素一一赋给nums1即可。
第三种:不用创建临时数组,也是采用双指针进行遍历,不过这次是从后往前遍历,依次将较大的元素放入nums1的尾部,这种方法最佳!
代码如下:
import java.util.Arrays;
import java.util.Scanner;
public class test25 {
/**
* 合并两个有序数组
*/
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
System.out.print("请输入第一个数组的大小:");
int M = sc.nextInt();
System.out.print("请输入第二个数组的大小:");
int N = sc.nextInt();
int[] nums1 = new int[M+N];
int[] nums2 = new int[N];
System.out.print("请输入第一个数组中的元素:");
for ( int i = 0; i < M+N; i++ ){
if ( i >= M ){
nums1[i] = 0;
}
nums1[i] = sc.nextInt();
}
System.out.print("请输入第二个数组中的元素:");
for ( int i = 0; i < N; i++ ){
nums2[i] = sc.nextInt();
}
// merge(nums1,M,nums2,N);
merge2(nums1,M,nums2,N);
System.out.println("合并后的nums1数组为:");
System.out.println(Arrays.toString(nums1));
}
private static void merge(int[] nums1, int m, int[] nums2, int n){
int left = 0, right = 0;
int[] temp = new int[m+n];
for ( int index = 0; index < m+n; index++ ){
if (left >= m){
while (right < n){
temp[index++] = nums2[right++];
}
break;
}
if (right >= n){
while (left < m){
temp[index++] = nums1[left++];
}
break;
}
if (nums1[left] < nums2[right]){
temp[index] = nums1[left];
left++;
}else {
temp[index] = nums2[right];
right++;
}
}
for ( int i = 0; i < m+n; i++ ){
nums1[i] = temp[i];
}
}
/**
* 这种方法采用倒序将元素放入nums1中,这样就不同申请多余的空间,达到节约空间的目的
* @param nums1
* @param m
* @param nums2
* @param n
*/
private static void merge2(int[] nums1, int m, int[] nums2, int n){
int l = m-1, r = n-1;
for ( int index = m+n-1; index >= 0; index-- ){
if ( l < 0 ){
while (r >= 0){
nums1[index--] = nums2[r--];
}
break;
}
if ( r < 0 ){
while (l >= 0){
nums1[index--] = nums1[l--];
}
break;
}
if (nums1[l] > nums2[r]){
nums1[index] = nums1[l];
l--;
}else {
nums1[index] = nums2[r];
r--;
}
}
}
}