[Leetcode] merge sorted array 合并数组

时间:2023-09-22 23:12:14

Given two sorted integer arrays A and B, merge B into A as one sorted array.

Note: 
You may assume that A has enough space to hold additional elements from B. The number of elements initialized in A and B are m and n respectively.

题意:给定两已排好的数组,将B合并到A中,A足够长。

思路:归并排序,只不过这题是要保存在A中,题目中的意思是不要开辟新的空间,所以常规的从左到右的归并排序行不通,我们可以从后面开始。这就涉及一个问题了:将数组A和数组B的最后元素相比较后,较大元素放在哪,即下标为多少?因,题中说,A足够大,所以至少m+n个位置,这样我们就可以将较大值放在下标为m+n-1处了。还有一个问题就是,若n>m,这样,主体的算法结束后,我们要考虑,n中的元素如何处理, 毫无疑问,此时m=0了,所以只需将B中的元素对应的赋给A中即可,代码如下:

 class Solution {
public:
void merge(int A[], int m, int B[], int n)
{
if(n==) return; while(m>&&n>)
{
if(A[m-]>B[n-])
{
A[m+n-]=A[m-];
m--;
}
else
{
A[m+n-]=B[n-];
n--;
}
}
while(n>)
{
A[m+n-]=B[n-]; //m不写也行,此时,m=0
n--;
}
}
};