题目理解
Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.
Note:
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2. The number of elements initialized in nums1 and nums2 are m and n respectively.
给两个有序数组num1,num2,合并为一个有序的数组存在num1中。
这里需要注意的是,合并之后数值存放在num1中,num1的长度大于等于num1.length()+num2.length()。
思路分析
本题的关键点就在于,合并之后的数组不新分配内存空间,而是利用已有的数组存放。
(1)所以从两个数组的尾部比较,大的后移到num1数组的m+n位置存放,并把被移动的数组指针往前移动。
(2)循环结束之后,判断num1、num2中是否有未比较移动的数,如果有则已经有序,循环遍历,放到index指针指向的位置之前部分。
(3)最后,所有数在num1中就是有序的了。如下图:
代码实现
/**
* 合并两个有序数组,并将合并之后的数据存放于已创建的数组内存空间中(不产生新的内存分配)
* @param nums1 有序数组1
* @param m
* @param nums2 有序数组2
* @param n
*/
public void merge(int[] nums1, int m, int[] nums2, int n) {
int i = m - 1; //数组1的元素个数
int j = n - 1; //数组2的元素个数
int index = m + n -1; //数组1的元素个数+数组2的元素个数-1 = 数组1的总长度。用以存储1和2合并之后的数据
while (i >= 0 && j >= 0) {
if (nums1[i] > nums2[j]) { //大的往后移动,指针往前移一个
nums1[index] = nums1[i];
index = index - 1;
i = i - 1;
} else {
nums1[index] = nums2[j];
index = index - 1;
j = j - 1;
}
}
//当上面循环结束后,num1或者num2可能还有数据没有进行比较/移动
//当num1中还有数据
while (i >= 0) {
nums1[index] = nums1[i];
index = index - 1;
i = i - 1;
}
//同上
while (j >= 0) {
nums1[index] = nums2[j];
index = index - 1;
j = j - 1;
}
}
测试
由于num1的实际长度应该大于等于连个数组数据个数之和,所以创建数组时,分配大一点的内存。
public static void main(String [] args) {
int[] num1 = new int[10];
num1[0] = 3;
num1[1] = 6;
num1[2] = 9;
int[] num2 = {10,11,12};
int m = 3;
int n = 3;
MergeTwoArray mt = new MergeTwoArray();
mt.merge(num1,m,num2,n);
for (int i = 0;i < num1.length;i++) {
System.out.println(num1[i] + " ");
}
}
小知识大道理,一点一滴积累。给点个赞吧
(个人微信公众号:TestOnTheRoad,搜索ID或者扫描下方二维码添加关注,关注测试开发工程师的成长之路)