题目:将两个排序好的数组组成一个新的排序好的数组,给出A=[1,2,3,4],B=[2,4,5,6],返回 [1,2,2,3,4,4,5,6]
挑战
你能否优化你的算法,如果其中一个数组很大而另一个数组很小?
思路: 两根指针分别指向两个数组的开头,每次取值两个指针位置较小的数,指针后移,直到其中一个数组结束,将另外一个数组指针后面的元素连接到新数组的后面即可
关于挑战的情况 如果A[0]>B[B.length-1]||A[A.length-1]<B[0] ,可根据情况 直接将A连接到B的后面(或者B连接到A的后面)
代码如下
public int[] mergeSortedArray(int[] A, int[] B) {
int i=0;
int j=0;
int res[] = new int[A.length+B.length];
int k=0;
while(i<A.length&&j<B.length){
if(A[i]<=B[j]){
res[k++]=A[i++];
}else{
res[k++]=B[j++];
}
}
if(i!=A.length){
for(int a=i;a<A.length;a++){
res[k++]=A[a];
}
}else{
for(int b=j;b<B.length;b++){
res[k++]=B[b];
}
}
return res;
}
}