【问题】
给定两个有序整数数组 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]
【思路】
这道题目的方法十分巧妙,由于题目中nums1有足够的空间,可以用来保存nums2的元素,将nums2合并到nums1,并且两者有序,这其实就是一个归并排序的方法。但巧妙之处在于如何调整合并后的顺序?
从后向前比较并放置较大的元素,如果nums1中前m个元素设置完成后,并且nums2有剩余,那么直接把剩下的元素拷贝到nums1中,这就是归并排序的思想!
【题解】
class Solution { public: void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) { int len = m+n-1; m = m-1, n = n-1; while(m >= 0 && n >= 0){ if(nums1[m] > nums2[n]){ nums1[len--] = nums1[m--]; }else{ nums1[len--] = nums2[n--]; } } while(n >= 0){ nums1[len--] = nums2[n--]; } } };