题目描述:给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明: 初始化 nums1 和 nums2 的元素数量分别为 m 和 n。你可以假设 nums1 有足够的空间来保存 nums2 中的元素。
示例:
输入: nums1 = [1,2,3,0,0,0], m = 3 , nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
解题思路:在数据结构课上学过有序链表的合并算法,所以在此借鉴该算法,首先需要创建一个新的数组res,同时遍历nums1和nums2两个数组,将两个数组中较小的元素赋给res,直至某个数组遍历完,接着将另一个未遍历完数组的剩余元素依次赋给res。
复杂度分析:
时间复杂度:O(m+n);空间复杂度:O(m+n)
下面贴出AC代码:
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int []res = new int[m+n];
int i=0,j=0,cnt=0;
//遍历两个数组,将较小的值依次赋给res数组
while(i<m&&j<n){
if(nums1[i]<nums2[j])
res[cnt++] = nums1[i++];
else
res[cnt++] = nums2[j++];
}
//将未遍历完的数组的剩余元素赋给res数组
if(i>=m)
while(j<n) res[cnt++] = nums2[j++];
if(j>=n)
while(i<m) res[cnt++] = nums1[i++];
//由于要返回的是nums1数组,所以需要将res数组的值赋给nums1
for(i=0;i<res.length;i++){
nums1[i] = res[i];
}
}
}