题目
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。
示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
示例 2:
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
思路
题干:
- 有序,nums1的大小为m+n
倒序遍历nums1和nums2,倒序网nuns1添加元素(从大到小逆序添加)
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
//nums1的实际元素个数尾下标
int i1=m-1;
//num2的尾下标
int i2=n-1;
//num1的尾下标
int i=m+n-1;
while(i1>=0&&i2>=0){
if(nums1[i1]>nums2[i2]){
nums1[i]=nums1[i1];
i--;
i1--;
}else{
nums1[i]=nums2[i2];
i--;
i2--;
}
}
//num2还未合并完就插入到nums1的首部
while(i2>=0){
nums1[i]=nums2[i2];
i--;
i2--;
}
}
}
时间复杂度O(m+n): m,n分别为nums1和nums2的元素个数
空间复杂度O(1) :常数大小