iOS常用算法之兩個有序數組合并（要求時間復雜度為0(n)）

2020-07-26 02:19:24

字體：大中小

來源：轉載

供稿：網友

思路:

常規思路:

先將一個數組作為合并后的數組, 然后遍歷第二個數組的每項元素, 一一對比, 直到找到合適的, 就插入進去;

　　簡單思路: 設置數組C, 對比A和B數組的首項元素, 找到最小的, 就放入數組C,依次進行下去.

代碼如下:

- (NSArray *)mergeOrderArrayWithFirstArray: (NSMutableArray *)array1 secondArray: (NSMutableArray *)array2 {  // 全為空不處理  if (!array1.count && !array2.count) {    return @[];  }  // 一個為空返回另外一個  if (!array1.count) {    return array2;  }  if (!array2.count) {    return array1;  }  NSMutableArray *endArray = [NSMutableArray array];  while (1) {    if ([array1[0] integerValue] < [array2[0] integerValue]) {      [endArray addObject:array1[0]];      [array1 removeObjectAtIndex:0];    }else {      [endArray addObject:array2[0]];      [array2 removeObjectAtIndex:0];    }    if (!array1.count) {      [endArray addObjectsFromArray:array2];      break;    }    if (!array2.count) {      [endArray addObjectsFromArray:array1];      break;    }  }  return endArray;}

執行代碼, 輸入輸出結果如下: