给定两个排序后的数组 A 和 B,其中 A 的末端有足够的缓冲空间容纳 B。 编写一个方法,将 B 合并入 A 并排序。
初始化 A 和 B 的元素数量分别为 m 和 n。
方法一:
双指针法:
代码:
class Solution {
public void merge(int[] A, int m, int[] B, int n) {
int[] result = new int[m+n];//造一个新的空间数组,用来存储排序后的数组
int pA = 0;
int pB = 0;
for(int i = 0; i < m+n; i++){
if(pA < m && pB < n){
if(A[pA] < B[pB]){
result[i] = A[pA++];
}else{
result[i] = B[pB++];
}
}else if(pA < m && pB == n ){
result[i] = A[pA++];
}else if(pA == m && pB < n){
result[i] = B[pB++];
}
}
for(int j = 0; j < m+n; j++){
A[j] = result[j];//将数组遍历放进A数组中
}
}
}
这道题的大概思路是,先创建一个合适大小(m+n)的新数组,然后用双指针法,分别指向A数组和B数组头部,由于是排序好的,所以可以依次遍历对比大小,将较小的数逐个放入新数组中,最后将新数组中的元素逐个放入A数组中。
方法二:
直接合并后排序法:
代码:
class Solution {
public void merge(int[] A, int m, int[] B, int n) {
int j = 0;
for(int i = m; i < m+n; i++){
A[i] = B[j++];
}
Arrays.sort(A);
}
}
其实第二种方法才是我第一时间想到的,就是直接合并到A数组里然后用工具sort排序,但是一想到在面试中用sort的同志经常都吃不了兜着走,所以我还是选择了使用双指针法,尽管实际测试下两种方法的时间和空间复杂度都不相上下。