合并两个有序数组[Leetcode]
混合插入有序数组,由于两个数组都是有序的,所以只要按顺序比较大小即可。题目中说了nums1数组有足够大的空间,说明我们不用resize数组,又给了我们m和n,那就知道了混合之后的数组大小,这样我们就从nums1和nums2数组的末尾开始一个一个比较,把较大的数,按顺序从后往前加入混合之后的数组末尾。需要三个变量i,j,k,分别指向nums1,nums2,和混合数组的末尾。进行while循环,如果i和j都大于0,再看如果nums1[i] > nums2[j],说明要把nums1[i]先加入混合数组的末尾,加入后k和i都要自减1;else,就把nums2[j]加入混合数组的末尾,之后j和k也要自减1。循环结束后,有可能i和j还大于等于0,若j大于等于0,那么需要继续循环,将nums2中的数字继续拷入nums1。若是i大于等于0,那么就不用管,因为混合数组本身就房子nums1中。
代码如下:
class Solution{ public void merge(int nums1 [], int m, int nums2 [], int n){ int i = m-1, j = n-1, k = m+n-1; while(i>=0 && j>=0){ if(nums1[i] > nums2[j]) nums1[k--] = nums1[i--]; else nums1[k--] = nums2[j--]; } while(j>=0){ nums[k--] = nums[j--]; } } }
还可以将两个while循环整合到一起,只要加上i > =0且nums1[i] > nums2[j]的判断条件就可以从nums1中取数:
class Solution { public void merge(int[] nums1, int m, int[] nums2, int n) { int i = m-1, j = n-1, k = m+n-1; while(j>=0){ nums1[k--] = i >= 0 && nums1[i]>nums2[j] ? nums1[i--] : nums2[j--]; } } }
nice .