灵感汇聚：揭秘前端刷题技巧——寻找两个正序数组的中位数

2023-12-01 17:22:49

前言

算法题作为前端面试的必备考察项目，其重要性不言而喻。然而，面对海量的算法题，如何高效地进行刷题练习，快速提升解题能力，成为前端工程师们亟需解决的问题。本文将以寻找两个正序数组的中位数这一经典算法题为例，详细讲解解题思路和步骤，帮助你轻松掌握前端刷题技巧，在面试中脱颖而出。

题目分析

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2，要求找出并返回这两个正序数组的中位数。

解题思路

解决此题的关键在于将两个有序数组合并为一个有序数组，然后从中提取中位数。具体步骤如下：

定义两个指针 i 和 j，分别指向数组 nums1 和 nums2 的起始位置。
比较 nums1[i] 和 nums2[j] 的值，较小的值加入合并后的数组中，同时将相应的指针向后移动一位。
重复步骤 2，直到两个数组中的所有元素都加入合并后的数组中。
合并后的数组中，中位数的位置为 (m + n) / 2。如果 (m + n) 为奇数，则中位数为合并后数组中下标为 (m + n) / 2 的元素；如果 (m + n) 为偶数，则中位数为合并后数组中下标为 (m + n) / 2 和 (m + n) / 2 - 1 的元素的平均值。

代码实现

/**
 * 寻找两个正序数组的中位数
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */
const findMedianSortedArrays = (nums1, nums2) => {
  const mergedArray = [];
  let i = 0;
  let j = 0;

  while (i < nums1.length && j < nums2.length) {
    if (nums1[i] < nums2[j]) {
      mergedArray.push(nums1[i]);
      i++;
    } else {
      mergedArray.push(nums2[j]);
      j++;
    }
  }

  while (i < nums1.length) {
    mergedArray.push(nums1[i]);
    i++;
  }

  while (j < nums2.length) {
    mergedArray.push(nums2[j]);
    j++;
  }

  const midIndex = Math.floor((nums1.length + nums2.length) / 2);

  if ((nums1.length + nums2.length) % 2 === 0) {
    return (mergedArray[midIndex] + mergedArray[midIndex - 1]) / 2;
  } else {
    return mergedArray[midIndex];
  }
};