leetcode寻找两个有序数组的中位数C++版
2019-05-11 23:12
603 查看
leetcode寻找两个有序数组的中位数
题目描述:
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。
示例 1:
nums1 = [1, 3] nums2 = [2] 则中位数是 2.0
示例 2:
nums1 = [1, 2] nums2 = [3, 4] 则中位数是 (2 + 3)/2 = 2.5
解决方案:
采用归并排序法:
- 判断nums1或nums2中是否为空,若其中一个为空,则直接判断另一个数组长度的奇偶性
- 如均不为空,则建立一个新数组temp,将nums1和nums2中数据依次比较大小并放入新数组temp中
- 如果其中一个数组比较完了,另一个数组还有剩余,则还要继续将这个数组放完
- 最后返回这个并归排序完的新数组,判断数组长度的奇偶性,得到结果
C++代码如下:
class Solution { public: double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) { int m = nums1.size(); int n = nums2.size(); if (nums1.empty()) { if (n%2 != 0) return 1.0*nums2[n/2]; return (nums2[n/2]+nums2[n/2-1])/2.0; } if (nums2.empty()) { if (m%2 != 0) return 1.0*nums1[m/2]; return (nums1[m/2]+nums1[m/2-1])/2.0; } int i = 0; int j = 0; vector<int> temp; while (i < m && j < n) { if (nums1[i] <= nums2[j]) { temp.push_back(nums1[i]); i++; } else { temp.push_back(nums2[j]); j++; } } if (i < m) { for (; i < m; i++) temp.push_back(nums1[i]); } else if (j < n) { for (; j < n; j++) temp.push_back(nums2[j]); } int len = temp.size(); if (len%2 != 0) return 1.0*temp[len/2]; return (temp[len/2]+temp[len/2-1])/2.0; } };
相关文章推荐
- Leetcode(C++)——4. 寻找两个有序数组的中位数
- [LeetCode]4. 寻找两个有序数组的中位数/C++
- Leetcode:4.寻找两个有序数组的中位数
- 【leetcode】4. 寻找两个有序数组的中位数
- LeetCode4.python实现:寻找两个有序数组中的中位数问题☆☆☆
- Python学习-Leetcode刷题记2:寻找两个有序数组的中位数
- LeetCode4 寻找两个有序数组的中位数 Median of Two Sorted Arrays
- @LeetCode寻找两个有序数组的中位数--Median of Two Sorted Arrays[C++]
- LeetCode--4. 寻找两个有序数组的中位数(Median of Two Sorted Arrays)
- LeetCode刷题笔记(Java版)----4. 寻找两个有序数组的中位数
- [leetcode] Median of Two Sorted Arrays 寻找两个有序数组的中位数
- leetcode 寻找两个有序数组的中位数
- 【LeetCode】第四题:寻找两个有序数组的中位数
- 基础算法之寻找两个有序数组的中位数(LeetCode-4)
- 寻找两个有序数组的中位数(LeetCode)
- LeetCode第4题--寻找两个有序数组的中位数
- LeetCode-4 :寻找两个有序数组的中位数 ★★★★★
- LeetCode题解(python)-4. 寻找两个有序数组的中位数
- leetcode 4.寻找两个有序数组的中位数
- leetcode--寻找两个有序数组的中位数