两个有序数组如何合并成一个有序数组

我这里考虑的两个数组均是升序排序,当然降序的两个数组进行合并算法是类似的。

下面有两段相似的代码,第一段除了返回合并后的有序数组还将这两个有序数组清空了,该算法的思路是始终比较两个数组的首元素大小,然后将小者 shift 出来 push 到结果数组中去,因为总是会将数组首元素较小的那个移出,故不用改变比较数组的索引值,一直固定为 0 就行了。最后不要忘记将长度值大于 0 的数组中的元素移出放置到结果数组中。

/* 清空了原来的两个有序数组 */
function mergeTwoSortedArr (arr1, arr2) {
  var retArr = [];
  /* 遍历比较两个数组的首元素大小,小者 shift 出来 push 到结果数组中去 */
  while (arr1.length > 0 && arr2.length > 0) {
    if (arr1[0] < arr2[0]) {
      retArr.push(arr1.shift());
    } else if (arr1[0] > arr2[0]) {
      retArr.push(arr2.shift());
    } else {
      retArr.push(arr1.shift());
      retArr.push(arr2.shift());
    }
  }
  /* 将数组(最多有一个)剩余元素移出放置到结果数组中 */
  while (arr1.length > 0) {
    retArr.push(arr1.shift());
  }
  while (arr2.length > 0) {
    retArr.push(arr2.shift());
  }
  return retArr;
}

// 示例
var arr1 = [2, 3, 5];
var arr2 = [3, 4, 7, 9];
console.log(mergeTwoSortedArr(arr1, arr2));   // [2, 3, 3, 4, 5, 7, 9]

第二段代码则没有影响原来的两个有序数组,通过遍历比较两个数组当前元素的大小,小者 push 到结果数组中去,相应数组索引加一,然后再进行循环比较。同样,最后不要忘记将未遍历过的数组元素复制到结果数组中。

/* 未对原来的两个有序数组做改动 */
function mergeTwoSortedArr (arr1, arr2) {
  var retArr = [];
  var len1 = arr1.length;
  var len2 = arr2.length;
  var i = 0, j = 0;
  /* 遍历比较两个数组当前元素的大小,小者 push 到结果数组中去,相应数组索引加一 */
  while (i < len1 && j < len2) {
    if (arr1[i] < arr2[j]) {
      retArr.push(arr1[i]);
      i++;
    } else if(arr1[i] > arr2[j]) {
      retArr.push(arr2[j]);
      j++;
    } else {
      retArr.push(arr1[i], arr1[i]);
      i++;
      j++;
    }
  }
  /* 将数组(最多有一个)剩余元素 push 到结果数组中 */
  for (; i < len1; i++) {
    retArr.push(arr1[i]);
  }
  for (; j < len2; j++) {
    retArr.push(arr2[j]);
  }
  return retArr;
}

// 示例
var arr1 = [2, 3, 5];
var arr2 = [3, 4, 7, 9];
console.log(mergeTwoSortedArr(arr1, arr2));   // [2, 3, 3, 4, 5, 7, 9]

注:上面两段代码算法的最后一步最多还要对一个数组进行操作——将其中的元素 push 到结果数组中去,因为之前的循环终止条件就是至少有一个数组已经遍历结束。

你可能感兴趣的:(两个有序数组如何合并成一个有序数组)