题目如下:
给出两个有序的整数数组 A和 B,请将数组 B合并到数组 A中,变成一个有序的数组
注意:可以假设 A数组有足够的空间存放B 数组的元素, A和B中初始的元素数目分别为 m和 n
解题思路:
- A、B数组均是有序数组,要把B数组的元素有序的存到A数组中,所以不能从A数组前面开始存要从后开始(如果从头开始存的话,会丢失A数组的元素)
- 将A数组与B数组中的元素比较,谁大谁就放到A数组的后面,直到其中一个数组元素轮完
- 然后将剩余元素存到A数组中
public class Solution {
public void merge(int A[], int m, int B[], int n) {
int a = m-1;
int b = n-1;
int sum = m+n-1;
while(a>=0 && b>=0){
if(A[a]>B[b]){
A[sum] = A[a];
a--;
sum--;
}else if(A[a]<=B[b]){
A[sum] = B[b];
b--;
sum--;
}
}
while(a >= 0){
A[sum] = A[a];
a--;
sum--;
}
while(b >= 0){
A[sum] = B[b];
b--;
sum--;
}
}
}