今天起更新leetcode「每日一题」的题
题干
给定两个排序后的数组 A 和 B,其中 A 的末端有足够的缓冲空间容纳 B。 编写一个方法,将 B 合并入 A 并排序。
初始化 A 和 B 的元素数量分别为 m 和 n。
示例:
输入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
想法
首先肯定第一个想法是把B的全部插入A的后边再sort
但是这样太麻烦 不要
接着想着双指针 两个指针指向两个数组的头,谁更小谁就先放进去
但是这样要存另一个数也没有很好使用已经排序好这个条件
于是想到从后边插入
双指针分别表示还有几个数字没有插入
从数组的尾巴开始比较,大的那个放到
m+n-1的位置
然后插入的那个数组指针左移
需要说明的是
可能出现m ,n中有一个已经为0 的情况
若是n为0 m不为0 证明数组A还没放完,但是比现在位置大的数都已经放好了
换言之只有A前m个数没放好,但是由于他们原本就是排序好的所以不用管
m为0 n不为0 那就证明只是B还有几个没放好,并且原本A[0]都大于B[n-1],那就剩下的都放上去就行
Java代码
class Solution {
public void merge(int[] A, int m, int[] B, int n) {
while(m>0&&n>0){
//大的先放 然后更新指针
if(A[m-1]>B[n-1]){
A[m+n-1]=A[m-1];
m--;
}
else{
A[m+n-1]=B[n-1];
n--;
}
}
//B还没放完的情况,A没放完不用处理
while(n>0){
A[n - 1] = B[n - 1];
n--;
}
}
}