题意: 给定两个排好序的数组A和B,将B合并到A中作为一个排好序的数组。
注意: 假设A有足够的空间来盛放A和B中合并的结果,初始时A和B中的元素个数分别是m和n。
分析: 因为数组对插入和删除操作非常不友好,因为需要移动后面的元素,相信大家都了解,因此我们在设计算法时都要避免直接对原数组的增删操作。我想到的方法是:因为我们事先已经直到A和B的大小分别是m和n,则合并后一定是m+n个元素,可以从后往前比较A和B中的元素,然后把大的放在后面,比如先比较A[m - 1] 和B[n - 1],将大的放在A数组的m+n-1处,以此类推。这么做可以避免直接在A数组中插入,并且能有效的解决该问题。
代码
public void merge(int A[], int m, int B[], int n) {
//从后往前比较两个数组
int i = m - 1; //i初始值是A的最后一个位置
int j = n - 1; //j初始值是B的最后一个位置
int count = 0; //表示已经处理了多少个数
while(i >= 0 && j >= 0){
if(A[i] >= B[j]){
A[m + n - 1 - count] = A[i];
i--;
count++;
}else{
A[m + n - 1 - count] = B[j];
j--;
count++;
}
}
while(j >= 0){ //A已经遍历完而B没有遍历完还需要将B中剩下的加入A中
A[m + n - 1 - count] = B[j];
j--;
count++;
}
//System.out.println(Arrays.toString(A));
}