给你两个按 非递减顺序 排列的整数数组 nums1
和 nums2
,另有两个整数 m
和 n
,分别表示 nums1
和 nums2
中的元素数目。
请你 合并 nums2
到 nums1
中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1
中。为了应对这种情况,nums1
的初始长度为 m + n
,其中前 m
个元素表示应合并的元素,后 n
个元素为 0
,应忽略。nums2
的长度为 n
。
示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3 输出:[1,2,2,3,5,6] 解释:需要合并 [1,2,3] 和 [2,5,6] 。 合并结果是 [1,2,2,3,5,6] ,其中斜体加粗标注的为 nums1 中的元素。
1.简单方法。使用Arrays.sort
public void merge(int[] nums1, int m, int[] nums2, int n) {
// 简单方法
for(int i =0;i<n;i++){
nums1[i+m] = nums2[i];
}
Arrays.sort(nums1);
}
2.双指针
1.创建一个新数组result,用来存储元素
2.分别遍历两个数组,当数组1的值>=数组2的值时 将数组2的值存储到数组result,数组2的指针index++,反之将数组1的值存储到result 数组1的指针index++
3.遍历result,赋值到nums1
注:需要判断数组index是否大于了数组长度,如果数组1index>数组1.length则剩余元素全部从数组二取,反之亦然。
代码:
public void merge(int[] nums1, int m, int[] nums2, int n) {
int all = m+n;
int[] result = new int[all];
for(int i = 0,nums1index = 0,nums2index = 0;i<all;i++){
if(nums1index>=m){
result[i] = nums2[nums2index++];
}else if(nums2index>=n){
result[i] = nums1[nums1index++];
}else if(nums1[nums1index]>=nums2[nums2index]){
result[i] = nums2[nums2index++];
}else{
result[i] = nums1[nums1index++];
}
}
for(int i = 0;i<all;i++){
nums1[i] = result[i];
}
}
3.优化双指针,不创建新数组。
代码
public void merge(int[] nums1, int m, int[] nums2, int n) {
int all = m+n;
for(int i = all -1,nums1Index = m-1,nums2Index = n-1;i>=0;i--){
if(nums1Index<0){
nums1[i] = nums2[nums2Index--];
}else if(nums2Index<0){
nums1[i] = nums1[nums1Index--];
}else if(nums1[nums1Index]<nums2[nums2Index]){
nums1[i] = nums2[nums2Index--];
}else{
nums1[i] = nums1[nums1Index--];
}
}
}