题目描述:
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
- 初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
- 你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:
输入: nums1 = [1,2,3,0,0,0], m = 3 nums2 = [2,5,6], n = 3 输出: [1,2,2,3,5,6]
解题思路:
解题思路就跟归并排序最后的merge思想一样的,用两个指针分别指向两个数组的头,然后就每次找一个最小的放到结果数组里去,然后该数所在的指针向后移动一个位置,直到某一个数组指针走到头了为止。
判断一下两个指针都到头了没有,如果没有的话就再把那个指针后面的数也弄过来放到结果数组里去,最后把数组整理到nums1中返回。
其实正确的思路应该是从后向前找,然后直接把结果在nums1中从后向前存,这样就不用浪费一个m+n的空间复杂度了。
代码实现(Java语言):
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int i = 0,j = 0;
int[] res = new int[m + n];
int temp = 0;
while(i < m && j < n){
if(nums1[i] <= nums2[j]){
res[temp++] = nums1[i++];
}else{
res[temp++] = nums2[j++];
}
}
while(i < m){
res[temp++] = nums1[i++];
}
while(j < n){
res[temp++] = nums2[j++];
}
for(i = 0;i<m+n;i++){
nums1[i] = res[i];
}
}
}