Median of Two Sorted Arrays
There are two sorted arrays nums1 and nums2 of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
https://leetcode.com/problems/median-of-two-sorted-arrays/
给定两个排序过的数组,取这两个数组合并后的中位数。
如果是奇数长度,中位数就是中间的数;如果是偶数长度,中位数是中间两个数的平均数。
难点是题目要求复杂度是O(m+n)。
我的解法是先算出两个数组的总长度,然后循环从排序过的两个数组中取比较小的那个,统计长度,当长度等于总长度的一半时,根据奇数和偶数两种情况得到结果。
1 /** 2 * @param {number[]} nums1 3 * @param {number[]} nums2 4 * @return {number} 5 */ 6 var findMedianSortedArrays = function(nums1, nums2) { 7 var i = 0, j = 0, count = 0, middleLeft = -1; 8 var len = nums1.length + nums2.length; 9 while(i < nums1.length || j < nums2.length){ 10 var curr = 0; 11 if(nums2[j] === undefined || nums1[i] <= nums2[j]){ 12 curr = nums1[i]; 13 }else if(nums1[i] === undefined || nums1[i] > nums2[j]){ 14 curr = nums2[j]; 15 } 16 count++; 17 if(len % 2 !== 0){ 18 if(count === (len - 1) / 2 + 1){ 19 return curr; 20 } 21 }else{ 22 if(count === len / 2){ 23 middleLeft = curr; 24 }else if(count === (len / 2 + 1) && middleLeft !== -1){ 25 return (curr + middleLeft) / 2; 26 } 27 } 28 29 if(nums2[j] === undefined || nums1[i] <= nums2[j]){ 30 i++; 31 }else if(nums1[i] === undefined || nums1[i] > nums2[j]){ 32 j++; 33 } 34 } 35 return 0; 36 };
时间: 2024-10-20 12:29:25