题目描述:
解题思路:(双指针法)
- 首先想到的是新建一个大小为(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;
}
}
}