[leetcode] Median of Two Sorted Arrays
来源:互联网 发布:苹果官方申请解网络锁 编辑:程序博客网 时间:2024/06/05 16:58
[4] 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)).
Example 1:
nums1 = [1,3]
nums2 = [2]
The median is 2.0
Example 2:
nums1 = [1,2]
nums2 = [3,4]
The median is (2+3)/2 = 2.5
给定两个排序好的数组,然后输出这两个数组的中数。
例如:
数组1 nums1 = [1,3]
数组2 nums2 = [2]
那么数组1和数组2的中数就是2.
再例如:
数组1 nums1 = [1,2]
数组2 nums2 = [3,4]
那么数组1和数组2的中数就是(2+3)/2=2.5。
解题思路
方法一:排序
这道题的困难度标记为Hard,题目很容易懂,使用对两个数组进行排序是一个非常容易想到的思路。
这里,我们可以对两个排序好的数组进行排序,整合成一个数组,然后取该数组的中数即可。
但是,这个有一个要特别注意的点:
若m+n为奇数,则中数(median)是排序好后的中间值;
若m+n为偶数,则中数(median)是排序好后的中间两个数的平均值。
也就是说,中数(median)不一定就是数组中的数字。
代码实现如下:
class Solution{public: double findMedianSortedArrays(vector<int>&nums1, vector<int>&nums2){ vector<int> mergeNums; mergeNums.assign(nums1.begin(), nums1.end()); mergeNums.insert(mergeNums.end(),nums2.begin(), nums2.end()); sort(mergeNums.begin(), mergeNums.end()); int len1 = nums1.size(); int len2 = nums2.size(); double median = (len1 + len2) % 2 ? mergeNums[(len1+len2)>>1] : (mergeNums[(len1+len2-1)>>1] + mergeNums[(len1+len2)>>1])/2.0; return median; }};
方法二:先删除,再查找
假设数组A和数组B的元素个数都大于k/2,我们将A的第k/2个元素(即A[k/2 -1])和B的第k/2的元素(即B[k/2 -1])进行比较,有以下三种情况(这里假设k为偶数,但是所得到的结论当k为奇数时,也是适用的):
1、A[k/2 -1] == B[k/2 -1]
2、A[k/2 -1] > B[k/2 -1]
3、A[k/2 -1] < B[k/2 -1]
如果A[k/2 -1] < B[k/2 -1],则A[0]到A[k/2 -1]肯定在A和B的top K元素的范围内,换句话说,A[k/2 -1]不可能大于A和B的第K大元素。
因此,我们可以放心的删除A的这k/2个元素。同理,当A[k/2 -1] > B[k/2 -1]时,可以删除B的k/2个元素。
当A[k/2 -1] == B[k/2 -1]时,说明找到了第k大元素,当直接返回A[k/2 -1]或者B[k/2 -1]。
因此,可以使用一个递归函数,就可以得到结果
1、当A或者B是空时,直接返回B[k-1]或者A[k-1];
2、当k=1时,返回min(A[0],B[0]);
3、当A[k/2 -1] == B[k/2 -1]时,返回A[k/2 -1]或者B[k/2 -1]。
代码实现如下:
double findKth(int a[], int m, int b[], int n, int k){ //always assume that m is equal or smaller than n if (m > n) return findKth(b, n, a, m, k); if (m == 0) return b[k - 1]; if (k == 1) return min(a[0], b[0]); //divide k into two parts int pa = min(k / 2, m), pb = k - pa; if (a[pa - 1] < b[pb - 1]) return findKth(a + pa, m - pa, b, n, k - pa); else if (a[pa - 1] > b[pb - 1]) return findKth(a, m, b + pb, n - pb, k - pb); else return a[pa - 1];}class Solution{public: double findMedianSortedArrays(int A[], int m, int B[], int n) { int total = m + n; if (total & 0x1) return findKth(A, m, B, n, total / 2 + 1); else return (findKth(A, m, B, n, total / 2) + findKth(A, m, B, n, total / 2 + 1)) / 2; }};
- LeetCode: Median of Two Sorted Arrays
- LeetCode Median of Two Sorted Arrays
- leetcode 26: Median of Two Sorted Arrays
- Leetcode 4 Median of Two Sorted Arrays
- [Leetcode] Median of Two Sorted Arrays
- LeetCode 4 - Median of Two Sorted Arrays
- [LeetCode]Median of Two Sorted Arrays
- leetcode Median of Two Sorted Arrays
- LeetCode-Median of Two Sorted Arrays
- [LeetCode] Median of Two Sorted Arrays
- [LeetCode] Median of Two Sorted Arrays
- [leetcode] Median of Two Sorted Arrays
- leetcode-004:Median of Two Sorted Arrays
- Leetcode 4 Median of Two Sorted Arrays
- leetcode之 median of two sorted arrays
- 【转载】【leetcode】Median of Two Sorted Arrays
- LeetCode - Median of Two Sorted Arrays
- [LeetCode]Median of Two Sorted Arrays
- Django第一个web程序
- Spring事务属性
- Linux权限控制的基本原理
- bootstrap 样式的左侧菜单树
- 智能指针类的实现
- [leetcode] Median of Two Sorted Arrays
- Git指令总结
- 一些网址记录
- Codeforces 754 D Fedor and coupons
- 【算法题】不等式数列
- java反射基础知识总结
- mysql 安装、建表及允许远程连接,及django连接mysql
- wlan没有有效的ip配置 win10
- mysql中合成索引的创建