LeetCode 4——两个排序数组中的中位数
1. 题目
2. 解答
2.1. 方法一
由于两个数组都是排好序的,因此首先可以想到的思路就是利用归并排序把两个数组合并成一个有序的长数组,然后直接取出中位数即可。
class Solution:
def findMedianSortedArrays(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: float
"""
len1 = len(nums1)
len2 = len(nums2)
nums = []
i = 0
j = 0
while (i < len1 and j < len2):
if (nums1[i] <= nums2[j]):
nums.append(nums1[i])
i += 1
else:
nums.append(nums2[j])
j += 1
if (i < len1):
while(i < len1):
nums.append(nums1[i])
i += 1
if (j < len2):
while(j < len2):
nums.append(nums2[j])
j += 1
odd = (len1 + len2) % 2 # 长度是否为奇数
mid = (len1 + len2 - 1) // 2
if (odd):
return nums[mid]
else:
return (nums[mid] + nums[mid+1]) / 2
因为要遍历两个数组,所以时间复杂度为 \(O(m+n)\),而题目中要求算法的时间复杂度为 \(O(log (m+n))\),因此应该是有更高效的算法,借助于二分查找。
2.2. 方法二
所谓中位数,就是将 K 个数据分为长度相等的两组,左边的数据小于等于右边的数据。
如果我们要在任意位置 \(i\) 处将长度为 \(m\) 的数组 \(A\) 分为两部分,则共有 \(m+1\) 种分法,\(i=[0, m]\)。
\]
\]
\(i=0\) 说明左半部分没有元素,反之 \(i=m\) 说明右半部分没有元素。左半边元素个数为 \(i\) ,右半边元素个数为\(m-i\)。
同理,我们可以在任意位置 \(j\) 处将长度为 \(n\) 的数组 \(B\) 分为两部分,则共有 \(n+1\) 种分法,\(j=[0, n]\)。
\]
针对划分完后的数组 \(A\) 和 \(B\),如果满足:
左边部分的长度等于右边部分的长度(偶数情况),\(i+j = m-i + n-j\);或者等于右边部分的长度加 1(奇数情况) ,\(i+j = m-i + n-j+1\)。
左边的最大元素小于等于右边的最小元素,\(A[i-1] <= B[j] \quad and \quad B[i-1] <= A[j]\)。
那我们也就找到了中位数,\(median = \frac{max(left\_part) + min(right\_part)}{2}\)。
所以,我们要做的就是在 \(i=[0, m]\) 区间搜索一个 \(i\) 值,使得上面的条件得到满足。也即
\]
加不加 1 取决于总的长度是奇数还是偶数,同时,为了保证 \(j\) 的范围在 \([0, n]\),我们必须要求 \(n\geqslant m\)。
接下来,我们就在 \(i=[0, m]\) 区间进行二分查找。
- 如果满足条件,则直接返回求取中位数。
- 如果 \(i > 0 \quad and \quad A[i-1] > B[j]\),则减小 \(i\)。如果增加 \(i\),则 \(j\) 减小,左边序列数组 \(A\) 的值会更大,右边序列数组 \(B\) 的值会更小。
- 如果 \(i < m \quad and \quad B[i-1] > A[j]\),则增加 \(i\)。如果减小 \(i\),则 \(j\) 增加,左边序列数组 \(A\) 的值会更小,右边序列数组 \(B\) 的值会更大。
最后,我们求得左半部分的最大值以及右半部分的最小值,然后就可以求出中位数了。
因为,要查找的范围为 \(i=[0, m]\),而且每次查找缩小区间为原来的一半,因此时间复杂度为 \(O(log(min(m, n))\),空间复杂度为 \(O(1)\)。
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int m = nums1.size();
int n = nums2.size();
int len = m + n;
int odd = len % 2;
int left = 0;
int i = 0, j = 0;
vector<int>::iterator A = nums1.begin();
vector<int>::iterator B = nums2.begin();
// 确保数组 A 的长度小于等于数组 B 的长度
if (m > n)
{
int temp = m;
m = n;
n = temp;
A = nums2.begin();
B = nums1.begin();
}
int right = m;
while(left <= right)
{
i = left + (right - left) / 2;
j = (len + odd) / 2 - i;
if (i > 0 && A[i-1] > B[j])
{
right = i - 1;
}
else if(i < m && B[j-1] > A[i])
{
left = i + 1;
}
else
{
break;
}
}
int left_max = 0;
int right_min = 0;
// 左半部分的最大值
if (i == 0) left_max = B[j-1];
else if (j == 0) left_max = A[i-1];
else left_max = A[i-1] <= B[j-1] ? B[j-1] : A[i-1];
// 右半部分的最小值
if (i == m) right_min = B[j];
else if (j == n) right_min = A[i];
else right_min = A[i] <= B[j] ? A[i] : B[j];
if (odd)
{
return left_max;
}
else
{
return float(left_max + right_min) / 2;
}
}
};
获取更多精彩,请关注「seniusen」!
LeetCode 4——两个排序数组中的中位数的更多相关文章
- LeetCode:寻找旋转排序数组中的最小值【153】
LeetCode:寻找旋转排序数组中的最小值[153] 题目描述 假设按照升序排序的数组在预先未知的某个点上进行了旋转. ( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0 ...
- 两个有序数组中的中位数以及求第k个最小数的值
解法参考 <[分步详解]两个有序数组中的中位数和Top K问题> https://blog.csdn.net/hk2291976/article/details/51107778 里面求中 ...
- 每日一道 LeetCode (8):删除排序数组中的重复项和移除元素
每天 3 分钟,走上算法的逆袭之路. 前文合集 每日一道 LeetCode 前文合集 代码仓库 GitHub: https://github.com/meteor1993/LeetCode Gitee ...
- leetcode 4.两个排序数组的中位数
题目: 给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 . 请找出这两个有序数组的中位数.要求算法的时间复杂度为 O(log (m+n)) . 你可以假设 nums1 和 nums ...
- 【LeetCode】34. 在排序数组中查找元素的第一个和最后一个位置
34. 在排序数组中查找元素的第一个和最后一个位置 知识点:数组,二分查找: 题目描述 给定一个按照升序排列的整数数组 nums,和一个目标值 target.找出给定目标值在数组中的开始位置和结束位置 ...
- [LeetCode] 154. 寻找旋转排序数组中的最小值 II
题目链接 : https://leetcode-cn.com/problems/find-minimum-in-rotated-sorted-array-ii/ 题目描述: 假设按照升序排序的数组在预 ...
- Java实现 LeetCode 154 寻找旋转排序数组中的最小值 II(二)
154. 寻找旋转排序数组中的最小值 II 假设按照升序排序的数组在预先未知的某个点上进行了旋转. ( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] ). 请找 ...
- Java实现 LeetCode 153 寻找旋转排序数组中的最小值
153. 寻找旋转排序数组中的最小值 假设按照升序排序的数组在预先未知的某个点上进行了旋转. ( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] ). 请找出其中 ...
- 领扣(LeetCode)寻找旋转排序数组中的最小值 个人题解
假设按照升序排序的数组在预先未知的某个点上进行了旋转. ( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] ). 请找出其中最小的元素. 你可以假设数组中不存在重 ...
随机推荐
- C# String与StringBuilder (转载)
1.什么时候用String?什么时候用StringBuilder? 字符串一旦创建就不可修改大小,所以对字符串添加或删除操作比较频繁的话.那就不要用String而用StringBuilder. 例如: ...
- Extjs treePanel 加载等待框
beforeload : { fn : function (store, operation, eOpts){ loadMask = new Ext.LoadMask(Ext.get(this.get ...
- SQL Server笔记-语法
1.USE <DatabaseName> //选择数据库 例:USE [master] //master是系统默认数据库 2.字段或表名与保留字或关键字重名时需要加. 3.COMPATIB ...
- Struts2 第五讲 -- Struts2与Servlet的API解耦
为了避免与 Servlet API 耦合在一起, 方便 Action 做单元测试, Struts2 对 HttpServletRequest, HttpSession 和 ServletContext ...
- linux 执行程序时,提示not found问题分析
sh: ./test: not found 通常可以通过readelf查看该进程文件所以依赖的运行环境,检查相关路径是否存在对应的文件. 比如如下: 1. 检查/lib目录,发现ld-X.XX.so为 ...
- 2018 Wannafly summer camp Day3--Shopping
Shopping 描述 题目描述: 你要买n件物品,其中有一些是凳子. 商场正在举行促销活动,如果购物车中有至少一个凳子,那么你可以半价购买这个购物车中最贵的一个物品. 你有m辆购物车,请最小化你的花 ...
- Super Reduced String
https://www.hackerrank.com/challenges/reduced-string/problem He wants to reduce the string to its sh ...
- SAP ABAP 日期,时间 相关函数
获的两个日期之间的分钟数 data min TYPE i. CALL FUNCTION 'DELTA_TIME_DAY_HOUR' EXPORTING T1 = ' T2 = ' D1 = ' D2 ...
- ABAP术语-Fiscal Year
Fiscal Year 原文:http://www.cnblogs.com/qiangsheng/archive/2008/02/02/1062109.html Generally a period ...
- 记js里codePointAt()方法返回的结果的含义。
经过<字符串的扩展>和<字符编码的那些事>这两篇文章的阅读,大概了解js里codePointAt方法返回结果的含义. var str='