题目介绍
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-sorted-array
解法一:
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int[] temp=new int[m];
for(int l=0;l<m;l++){
temp[l]=nums1[l];
}
int i=0,j=0;
int k=0;
while(i<m&&j<n){
if(temp[i]<nums2[j]){
nums1[k]=temp[i];
k++;
i++;
}
else if(temp[i]>nums2[j]){
nums1[k]=nums2[j];
k++;
j++;
}
else{
nums1[k]=temp[i];
k+=1;
nums1[k]=temp[i];
k++;
i++;
j++;
}
}
while(i<m){
nums1[k]=temp[i];
i++;
k++;
}
while(j<n){
nums1[k]=nums2[j];
j++;
k++;
}
}
}
结果:
时间复杂度:O(m+n)
空间复杂度:O(m)
注意事项:
在将值保存到另一个数组中时
int[] temp=nums1;这样赋值是不行的。
解法二:
将数组二添加到数组一尾部,重新进行排序。
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
for (int i = 0; i != n; ++i) {
nums1[m + i] = nums2[i];
}
Arrays.sort(nums1);
}
}
时间复杂度:O((m+n)\log(m+n))O((m+n)log(m+n))。
排序序列长度为 m+nm+n,套用快速排序的时间复杂度即可,平均情况为 O((m+n)\log(m+n))O((m+n)log(m+n))。
空间复杂度:O(\log(m+n))O(log(m+n))。
排序序列长度为 m+nm+n,套用快速排序的空间复杂度即可,平均情况为 O(\log(m+n))O(log(m+n))。
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/merge-sorted-array/solution/he-bing-liang-ge-you-xu-shu-zu-by-leetco-rrb0/
来源:力扣(LeetCode)
解法三:
方法一需要进行赋值是因为数组中的元素可能会被覆盖。观察发现在数组1中的后边的元素是无意义的,可以进行赋值而不用担心被覆盖。所以可以从后边开始比较,将比较较大的放在后边。
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int i=m-1;
int j=n-1;
int tail=m+n-1;
while(i>=0&&j>=0){
if(nums1[i]<nums2[j]){
nums1[tail]=nums2[j];
j--;
tail--;
}else if(nums1[i]>nums2[j]){
nums1[tail]=nums1[i];
i--;
tail--;
}else{
nums1[tail]=nums1[i];
nums1[--tail]=nums1[i];
tail--;
i--;
j--;
}
}
while(j>=0){
nums1[tail]=nums2[j];
j--;
tail--;
}
}
}
这样空间复杂度就变为了O(1)
同时官方给出了更为简洁的代码
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int p1 = m - 1, p2 = n - 1;
int tail = m + n - 1;
int cur;
while (p1 >= 0 || p2 >= 0) {
if (p1 == -1) {
cur = nums2[p2--];
} else if (p2 == -1) {
cur = nums1[p1--];
} else if (nums1[p1] > nums2[p2]) {
cur = nums1[p1--];
} else {
cur = nums2[p2--];
}
nums1[tail--] = cur;
}
}
}
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/merge-sorted-array/solution/he-bing-liang-ge-you-xu-shu-zu-by-leetco-rrb0/
来源:力扣(LeetCode)