题解 | #合并两个有序的数组#
合并两个有序的数组
http://www.nowcoder.com/practice/89865d4375634fc484f3a24b7fe65665
此文仅限于本人学习记录
小小的简单题居然有大大的坑。首先想要表示A的实际长度那里就不能用A.length而是m,因为题目说了A实际有效值是m,可是长度被设定为m+n。然后就是[],[1]这种极端情况需要直接复制B数组。还有就是java里面是不能数组A = 数组B这种方式来将数组B复制到A的。
public class Solution { public void merge(int A[], int m, int B[], int n) { if(m == 0){ //用于应付[],[1]这种情况 for(int i = 0;i < n;i++){ A[i] = B[i]; } return; } int[] temp = new int[m+n]; int aIter = 0; int bIter = 0; for(int i = 0;i < m+n;i++){ if(aIter >= m){ ////注意这里不能写A.length while(bIter < B.length){ temp[i++] = B[bIter++]; } break; } else if(bIter >= B.length){ while(aIter < m){ //注意这里不能写A.length,因为A数组长度不是m,只有m个有效元素但是长度被设定为m+n temp[i++] = A[aIter++]; } break; } if(A[aIter] < B[bIter]){ temp[i] = A[aIter++]; }else{ temp[i] = B[bIter++]; } } for(int i = 0;i < m+n;i++){ A[i] = temp[i]; } } }