给你两个按 非递减顺序 排列的整数数组 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 中的元素。
示例 2:
输入:nums1 = [1], m = 1, nums2 = [], n = 0 输出:[1] 解释:需要合并 [1] 和 [] 。 合并结果是 [1] 。
示例 3:
输入:nums1 = [0], m = 0, nums2 = [1], n = 1 输出:[1] 解释:需要合并的数组是 [] 和 [1] 。 合并结果是 [1] 。 注意,因为 m = 0 ,所以 nums1 中没有元素。nums1 中仅存的 0 仅仅是为了确保合并结果可以顺利存放到 nums1 中。
提示:
nums1.length == m + n
nums2.length == n
0 <= m, n <= 200
1 <= m + n <= 200
-109 <= nums1[i], nums2[j] <= 109
进阶:你可以设计实现一个时间复杂度为 O(m + n)
的算法解决此问题吗?
一道入门的简单题,题目并不难,最简单就直接把和数组合并一个sort排序,但是算法重要的思维,实际上按照他给的时间复杂度,其实就是提示一次遍历就能实现,那么其实解法就出来了,运用指针手动模拟遍历数组,每次存放两个数组中的较小的那一个,因为题目有个前置条件, 非递减顺序,那么接下来就开始实现。
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
//创建临时数组来存储合并后的的结果
int[] temp = new int[m+n];
//初始化两个数组的索引
int a=0,b=0,sum=0;
//当两个数组都未遍历完时,小的先入
while(a<m && b<n){
if(nums1[a] < nums2[b]){
temp[sum++] = nums1[a++];
}
else {
temp[sum++] = nums2[b++];
}
}
//有一个数组遍历完,直接将另外一个数组剩余全部加进来
while (a<m){
temp[sum++] = nums1[a++];
}
while (b<n)
temp[sum++] = nums2[b++];
//运用自带的复制将temp复制给nums1,底层实现还是遍历赋值,也可以手写遍历赋值
System.arraycopy(temp,0,nums1,0,m+n);
}
};
}
其实实现还是很简单的,接下来创建测试类,验证程序
public static void main(String[] args) {
Solution solution = new Solution();
int nums1[]={1,2,3,0,0,0};
int m = 3;
int nums2 []= {2,5,6};
int n = 3;
solution.merge(nums1,m,nums2,n);
for (int i : nums1) {
System.out.print(i+" ");
}
}
运行结果
ok,简单测试完成进行最后的提交判定
其实这道题一开始用for写,写到最后发现手动复制数组太麻烦,改成while就没有for那么臃肿,if嵌套也少了很多。