题目描述:
解题思路:(双指针法)
- 首先想到的是新建一个大小为(m+n)的数组,用来临时存储排好序的数组,最后将其复制到
nums1
中。- 也考虑到是否可以原地算法,不开辟新的空间,直到看到题解,从后向前进行排序,从大到小才恍然大悟。
nums1
中后面的n位全是0,是可以被覆盖的。- 注意:对于逆向双指针法,对于
index2 < 0
的情况,直接返回即可,此时说明nums2
已经全部放置在了nums1
中。
代码:
class Solution { public void merge(int[] nums1, int m, int[] nums2, int n) { int p1 = 0, p2 = 0; int[] sorted = new int[m + n]; int cur; while (p1 < m || p2 < n) { if (p1 == m) { cur = nums2[p2++]; } else if (p2 == n) { cur = nums1[p1++]; } else if (nums1[p1] < nums2[p2]) { cur = nums1[p1++]; } else { cur = nums2[p2++]; } sorted[p1 + p2 - 1] = cur; } for (int i = 0; i != m + n; ++i) { nums1[i] = sorted[i]; } } }
//逆向双指针法 class Solution { public void merge(int[] nums1, int m, int[] nums2, int n) { int index1 = m - 1; int index2 = n - 1; int cur; int tail = m + n - 1; while (index1 >= 0 || index2 >= 0){ if (index1 < 0){ cur = nums2[index2--]; }else if(index2 < 0){//此种情况,说明Nums2已经全部放置好,不需要继续排队直接返回即可 return; //cur = nums1[index1--]; }else if(nums1[index1] <= nums2[index2]){ cur = nums2[index2--]; }else { cur = nums1[index1--]; } nums1[tail--] = cur; } } }